使用自己的数据跑通mmsegmentation

目录

0.安装必要的库

1.加载数据

2.backbone选择

1)预训练模型的下载

2)设置最大训练epoch数，以及多少个epoch训练一次

3.优化器选择

4.训练

1）单卡训练

2）多卡训练

4.loss+结果回传

5.结果

报错

1）type object 'FileClient' has no attribute 'infer_client'

2）It is expected output_size equals to 2, but got size 3

3）训练测试结果报NAN

4）KeyError: 'MyDataset is not in the dataset registry'

5）预训练模型和加载的网络结构层名有差异,unexpected key in source state_dict:

6）使用多卡训练时报错：tools/dist_train.sh：Bad substitutation

7)results['ann_info']['seg_map'])  KeyError: 'ann_info

---

0.安装必要的库

pip install mmsegmentation-0.24.1-py3-none-any.whl

pip install mmcv-full==1.4.4 -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html

注：先查看当前 mmsegmentation对应的mmcv-full版本，在README_zh-CN.md中有提到当前的mmsegmentation版本，然后通过mmsegmentation版本查找mmcv-full版本，在docs/zh_cn/get_started.md中有对应信息：

然后查看cuda版本和pytorch版本，查看cuda版本使用nvcc -V,查看pyotch版本使用conda list