Openmmlab寒假训练营(七)

Openmmlab寒假训练营(七)

MMSegmentation使用小结

官方代码仓库链接：MMSegmentation官方

训练公开数据集

1.下载公开数据集
2.配置自己的config文件
训练模型配置+数据配置+学习率、优化器schedule策略配置+default_run_time配置
3.按照需求修改自己的训练config
4.开始训练

训练自定义数据集

1.使用如LabelMe标注自己ide数据集，并制作对应的label
2.利用@DATASET.register_module()写入自己的数据集类别，其中必须包含CLASSES=(…)和PALETTE=[[],[],…,[]]，再将自己的数据集添加到"mmseg/datasets/init.py"中
3.配置自己训练的config（和上述公开数据集训练类似）

注：一些具体的过程可以参考 MMSegmentation<1.0使用指南