bubbliiiing / segformer-pytorch

这是一个segformer-pytorch的源码,可以用于训练自己的模型。
MIT License
271 stars 35 forks source link

这个模型只对多类别的复杂的数据集支持较好吗? #32

Open ZJDATY opened 5 months ago

ZJDATY commented 5 months ago

我的数据集是3000张单标签的,只针对单类的数据,即numclasses=2 。 效果惨不忍睹,和大佬复现的那个deeplabv3+模型,miou差了10-20,相同数据集,相同参数下,deeplabv3+ ,我训练出来的的miou可以到99.5,背景的iou是99.9,标签的iou是99。但是这个segformer我测试了b0 和b2 ,b2稳定下来是miou 是90.9,背景是iou是98.85,标签iou只有84.0。看了mask的叠加图,都错位了。然后又测了b0,好家伙,miou到80多就升不上去了。@bubbliiiing 哥 能不能稍微解惑一下。我前几天也测试了 https://github.com/OpenGVLab/InternImage 这里面的 InternImage+Mask2Former ,使用的是dcn,现在dcn v4 也出来了,大佬有没有兴趣搭建一下。 image

image

ZJDATY commented 5 months ago

我的问题,输入shape 写反了。但还是希望大佬可以搞一下dcnv4。@bubbliiiing