deeplabv3+语义分割模型训练

我的一个项目是关于deeplabv3+的语义分割模型,目的是复现别人的结果,用的是电池负极外观的一个数据集,数据集450多张,train val比8:2,现在一个我的训练效果,iou,miou都特别低,复现标准为下面的图

img

我的结果

img

由于我是刚接触这方面,不知道专业术语的表达,我把我的工作用通俗语言表达一下
数据预处理
我先用labelme脚本生成生成灰度图,然后把不同mask值分成不同区域如下图,我有5个标签,但其中第一个标签归为背景一类

img

img

然后分成训练集与数据集,再计算每个标签的权重,之后再分割图片大小为512*512。我的预处理工作就做完了

接下来我用全监督训练,那些什么网络代码我都有,我也不知道怎么去检查那些网络代码的正确性,我的把gpu,epoch,batchsize都设置好,然后用的resnet18那个网络我也在pytorch官网下了一个,我就开始训练了,然后训练中间两个标签怎么都训练不出来。
我现在也没有什么想法,有没有人可以提供一点思路看看是什么地方可能出现问题,感激不尽。

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^