Skip to content
Extraits de code Groupes Projets
Valider 4f3f2647 rédigé par Victor Joos de ter Beerst's avatar Victor Joos de ter Beerst
Parcourir les fichiers

Segmentation task

parent e557f2a0
Aucune branche associée trouvée
Aucune étiquette associée trouvée
Aucune requête de fusion associée trouvée
_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
_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
# @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
_target_: flash.image.SemanticSegmentation
backbone: "mobilenetv3_large_100"
head: fpn
num_classes: ${datamodule.num_classes}
complete: []
defaults:
- download
url: "https://github.com/ongchinkiat/LyftPerceptionChallenge/releases/download/v0.1/carla-capture-20180513A.zip"
0% Chargement en cours ou .
You are about to add 0 people to the discussion. Proceed with caution.
Terminez d'abord l'édition de ce message.
Veuillez vous inscrire ou vous pour commenter