Customizable data pipeline for object detection about lightning-flash HOT 3 CLOSED

@reactivetype DataPipeline is already merged. Please, check if that suits your use case. On the other hand, we are refactoring the data modules to make it more flexible and user friendly in front of custom data structures. Take a look at #256

from lightning-flash.

Hi, @reactivetype! Yes, that sounds great. As you can see currently, the flow for OD is like this:

datamodule = ObjectDetectionData.from_coco(
    train_folder="data/coco128/images/train2017/",
    train_ann_file="data/coco128/annotations/instances_train2017.json",
    batch_size=5
)

model = ObjectDetector(num_classes=datamodule.num_classes)

We could add support for more datasets by adding class methods to the ObjectDetectionData class. For eg., ObjectDetectionData.from_yolo(..), ObjectDetectionData.from_voc(..), etc.

Yes, you could pass transformations functions to the train_transform argument in ObjecDetectionData.from_coco.

The purpose of the DataPipeline is to provide the flow for the transformation of data using hooks. So, depending on your data requirements, you could tweak it by creating a Subclass of it.

But right now, we are doing a refactor on DataPipeline #141. Hence, the behavior could change but would be a better experience for the User! :)

from lightning-flash.

Please feel free to reopen if needed!

from lightning-flash.