deeplabv3+pytorch补充教程

主要参考：

https://blog.csdn.net/qq_39056987/article/details/106455828?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522163283720916780265498868%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=163283720916780265498868&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2_allfirst_rank_ecpm_v1~rank_v31_ecpm-1-106455828.pc_search_result_cache&utm_term=deeplabv3++pytorch&spm=1018.2226.3001.4187

这篇博客已经记录的很全了，我再补充一点，我自己粗心遇到的问题。

图片格式的问题

文件：dataset（自己的名称）.py

自己制作的图像分割的数据集，图片和mask都是png格式的，需要再在数据集的py文件中修改拓展名，在如下的位置，我是用的数据集是sis，就是在sis.py，如果有人把图片直接放在放在了coco或pascal的路径中，就去相应的coco.py或pascal.py中去修改。

数据集的类别也千万别忘了在这个文件中修改

数据集没安全修改的问题（改了但没完全改）

文件：init.py

参照之前的博客，这部分自己的数据集是和pascal格式一样的，于是把上面pascal中有用的部分复制过来即可。

但是漏改了图中红色框的两个小名称，复制过来是pascal，也应改为自己数据集起的英文名，才可以正确找到类别等。

utils.py文件设置好类别和颜色

看图即可，和原博文一样，不再展开

train.py中设置好数据集名称和路径，以及loss类型

看图即可，和原博文一样，不再展开

推理.py文件（有的叫demo，有的叫infer，有的叫predict）

需要注意这个文件要改类别！不然会报错，报错去搜，有的人会说需要重新装cudnn什么的，其实不用，只是推理的时候，类别没设置好。

同时也可以设置生成mask的输出路径

最后程序训练的时候，是这个样子的