bubbliiiing/unet-pytorch

关于二分类数据集制作问题

greatvalley opened this issue · 1 comments

up你好,我目前是做二分类,然后用labelme标了150张图片,结果边缘的效果不是很好,大体上还是能够识别出来,然后我想用opencv做掩膜出来,目前已经把掩膜分割出来了,但是没办法用这个unet网络进行训练,训练出来都是全黑的,
image
想问问up有什么办法可以把我用opencv做的这个掩膜转换成voc格式的标签图,我试过把前景的颜色转成一样的,但实际上用不了,谢谢up

如果用右边这个训练全是0.5的准确率,最后效果很差