Segmentation task

hydra_flash/conf/datamodule/lyft.yaml

+_target_: flash.image.SemanticSegmentationData.from_folders
+train_folder: "${preprocess.path}/CameraRGB"
+train_target_folder: "${preprocess.path}/CameraSeg"
+val_split: 0.1
+transform_kwargs:
+  image_size: [256, 256]
+num_classes: 21
+batch_size: 4

hydra_flash/conf/datamodule/lyft_test.yaml

+_target_: flash.image.SemanticSegmentationData.from_files
+predict_files:
+  - "${preprocess.path}/CameraRGB/F61-1.png"
+  - "${preprocess.path}/CameraRGB/F62-1.png"
+  - "${preprocess.path}/CameraRGB/F63-1.png"
+batch_size: 3

hydra_flash/conf/experiment/lyft.yaml

+# @package _global_
+defaults:
+  - _self_
+  - /server: local
+  - /preprocess: lyft
+  - /datamodule: lyft
+  - /model: seg_mobilenet
+  - /datamodule@predict_datamodule: lyft_test
+
+trainer:
+  _target_: flash.Trainer
+  max_epochs: 3
+  gpus: 1

hydra_flash/conf/model/seg_mobilenet.yaml

+_target_: flash.image.SemanticSegmentation
+backbone: "mobilenetv3_large_100"
+head: fpn
+num_classes: ${datamodule.num_classes}
+complete: []

hydra_flash/conf/preprocess/lyft.yaml

+defaults:
+  - download
+
+url: "https://github.com/ongchinkiat/LyftPerceptionChallenge/releases/download/v0.1/carla-capture-20180513A.zip"