TypeError: __init__() got an unexpected keyword argument 'transform'

🐛 Bug

Uncaught exception
Traceback (most recent call last):
  File "/mnt/data/buduanban_detection/meteo.py", line 103, in <module>
    datamodule = ObjectDetectionData.from_coco(
  File "/data/miniconda3/envs/torch/lib/python3.9/site-packages/flash/image/detection/data.py", line 386, in from_coco
    return cls.from_icedata(
  File "/data/miniconda3/envs/torch/lib/python3.9/site-packages/flash/image/detection/data.py", line 220, in from_icedata
    return cls(
TypeError: __init__() got an unexpected keyword argument 'transform'

To Reproduce

The code is directly copied from https://lightning-flash.readthedocs.io/en/latest/reference/object_detection.html

Code sample

from dataclasses import dataclass
import albumentations as alb
from icevision.tfms import A

from flash import InputTransform
from flash.core.integrations.icevision.transforms import IceVisionTransformAdapter
from flash.image import ObjectDetectionData


@dataclass
class BrightnessContrastTransform(InputTransform):
    image_size: int = 128

    def per_sample_transform(self):
        return IceVisionTransformAdapter(
            [*A.aug_tfms(size=self.image_size), A.Normalize(), alb.RandomBrightnessContrast()]
        )


datamodule = ObjectDetectionData.from_coco(
    train_folder="data/coco128/images/train2017/",
    train_ann_file="data/coco128/annotations/instances_train2017.json",
    val_split=0.1,
    transform=BrightnessContrastTransform,
    batch_size=4,
)

Expected behavior

Environment

• OS (e.g., Linux):
• Python version:
• PyTorch/Lightning/Flash Version (e.g., 1.10/1.5/0.7):
• GPU models and configuration:
• Any other relevant information:

Additional context

[ligaz]

There is a breaking change in the latest version of flash that unifies all transforms in a single API implemented in https://github.com/PyTorchLightning/lightning-flash/pull/1233. Please refer to the 0.7.1 version of the docs in order to make this work - https://lightning-flash.readthedocs.io/en/0.7.1/reference/object_detection.html

In your particular case you should use train_transform instead of transform (the new API) in ObjectDetectionData.from_coco call.

[IncubatorShokuhou]

There is a breaking change in the latest version of flash that unifies all transforms in a single API implemented in #1233. Please refer to the 0.7.1 version of the docs in order to make this work - https://lightning-flash.readthedocs.io/en/0.7.1/reference/object_detection.html In your particular case you should use train_transform instead of transform (the new API) in ObjectDetectionData.from_coco call.

Thank you @ligaz , it works.

If you are happy with it, please feel free to close this issue 😃 Thank you for your patience. 😃

Thank you @krshrimali.