虽然做息肉分割除了kvasir-seg也没有太多好的选择,但这个数据集的问题挺多的,不能直接使用。相比其他领域的数据集,这个数据集真的很粗糙。
首先,需要二值化
kvasir-seg数据集作为一个只有单一前景的分割任务,按理说其mask只能是二值化的,但是现在却有不止两种值。其黑色区域的值可能是0, 1, 2, 3, 4, 5, 6, 7,8;白色区域可能是48, 249, 250, 251, 252, 253, 254, 255等。需要先做二值化。
其次,维度问题,要注意,最后输入训练的mask的channel必须是1.
第三,要注意用torch.FloatTensor将读取的图片数据转换为float类型,否则会报错