diff --git a/hydra_flash/conf/datamodule/lyft.yaml b/hydra_flash/conf/datamodule/lyft.yaml new file mode 100644 index 0000000000000000000000000000000000000000..26e2b3b6b8cf213fec364de9bea3df494986aa1c --- /dev/null +++ b/hydra_flash/conf/datamodule/lyft.yaml @@ -0,0 +1,8 @@ +_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 diff --git a/hydra_flash/conf/datamodule/lyft_test.yaml b/hydra_flash/conf/datamodule/lyft_test.yaml new file mode 100644 index 0000000000000000000000000000000000000000..ea649de6124940128d144003239e2f6ec4503f1d --- /dev/null +++ b/hydra_flash/conf/datamodule/lyft_test.yaml @@ -0,0 +1,6 @@ +_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 diff --git a/hydra_flash/conf/experiment/lyft.yaml b/hydra_flash/conf/experiment/lyft.yaml new file mode 100644 index 0000000000000000000000000000000000000000..b12d382a4570f47721bfb4aacbfc8643fd7d2cfc --- /dev/null +++ b/hydra_flash/conf/experiment/lyft.yaml @@ -0,0 +1,13 @@ +# @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 diff --git a/hydra_flash/conf/model/seg_mobilenet.yaml b/hydra_flash/conf/model/seg_mobilenet.yaml new file mode 100644 index 0000000000000000000000000000000000000000..bc512ff512ae3534cd35ba99290663b7822615c9 --- /dev/null +++ b/hydra_flash/conf/model/seg_mobilenet.yaml @@ -0,0 +1,5 @@ +_target_: flash.image.SemanticSegmentation +backbone: "mobilenetv3_large_100" +head: fpn +num_classes: ${datamodule.num_classes} +complete: [] diff --git a/hydra_flash/conf/preprocess/lyft.yaml b/hydra_flash/conf/preprocess/lyft.yaml new file mode 100644 index 0000000000000000000000000000000000000000..63a8fe6a4bdf656fda3c5854bf2a2fa651b30358 --- /dev/null +++ b/hydra_flash/conf/preprocess/lyft.yaml @@ -0,0 +1,4 @@ +defaults: + - download + +url: "https://github.com/ongchinkiat/LyftPerceptionChallenge/releases/download/v0.1/carla-capture-20180513A.zip"