https://kgithub.com/open-mmlab/mmsegmentation
1.下载mmcv
pip install -U openmim
mim install mmcv-full
2.手动下载代码到本地或者使用git clone https://github.com/open-mmlab/mmsegmentation.git
3.进入mmdetection目录:cd mmsegmentation
4.pip install -v -e .
如需使用 下载对应版本的mmcv
eg:
mim install mmcv-full==1.7.0
数据可以使用labelme进行数据标注,labelme还提供了数据集格式转换脚本,可以将labelme数据集格式转换为voc数据集格式
转换后:
JPEGImages为图片,SegmentationClassPNG为标签
1.运行train.py,生成完整的模型配置文件,对完整的模型配置文件进行修改。
配置文件每个GPU的batch_size一定要大于1