custom dataset training for segmentation

[soheilmajidi]

hi is there any way to train segformer for custom dataset with pretrain weights? thanks

[FatemaD1]

Hi, how to create data in the format required by the segformer model? @soheilmajidi

[NowLoadY]

same questions too... Did anyone solved ?😵‍💫

[QuietWoods]

参考 docs/tutorials/customize_datasets.md

1.在 local_configs/datasets 仿照cityscapes.py写一个自定义数据集文件。 里面的dataset_type改成CustomDataset。data_root写你自己的数据集路径。

dataset_type = 'CustomDataset'

2. 仿照local_configs/segformer/B1/segformer.b1.512x512.ade.160k.py，写一个自定义配置文件。

before: ` base = [ '../../base/models/segformer.py', '../../base/datasets/ade20k_repeat.py', '../../base/default_runtime.py', '../../base/schedules/schedule_160k_adamw.py' ]

num_classes=150,

` after:

` base = [ '../../base/models/segformer.py', '../../base/datasets/custom.py', '../../base/default_runtime.py', '../../base/schedules/schedule_160k_adamw.py' ]

num_classes=`