mmsegmentation运行自己的VOC数据集

参考:https://blog.csdn.net/weixin_43570470/article/details/124183068

1.克隆mmsegmentation

git clone https://github.com/open-mmlab/mmsegmentation.git

2.拷贝configs文件夹,防止破坏原来的configs文件

mmsegmentation运行自己的VOC数据集_第1张图片

3.修改configs/deeplabv3plus/deeplabv3plus_r50_d8_20k_voc文件

mmsegmentation运行自己的VOC数据集_第2张图片

4.修改configs/_base/models/deeplabv3plus_r50-d8.py

主要修改num_classes,注意有两处num_classes。num_classes为类别个数+1
mmsegmentation运行自己的VOC数据集_第3张图片

5.修改configs/_base/database/pascal-voc12.py

修改成你自己的数据集位置
mmsegmentation运行自己的VOC数据集_第4张图片

6.修改mmseg/datasets/voc.py

修改成你自己的类别和对应的rgb值
mmsegmentation运行自己的VOC数据集_第5张图片

7.修改mmseg/core/evalution/class_names.py

修改成自己的类别
mmsegmentation运行自己的VOC数据集_第6张图片

8.运行

在这里插入图片描述

9.运行结束后会生成一个workdirs文件里面保存训练模型

你可能感兴趣的:(深度学习,tensorflow,人工智能)