语义分割数据集组织



我正在尝试用语义分割来分割4个病变。我遵循这个这个伟大的后

我的训练文件夹中只有2个子文件夹带有补丁:遮罩和图像。在带有掩码的文件夹中,所有类都是混合的。另一个文件夹具有相应的图像。所以,当我训练模型时,它出现了:一个类被发现,就在上面提到的帖子后面。结果令人失望,我想知道是否必须在文件夹中拆分类,因此模型识别出4个类而不是一个。

您真正需要注意的是创建遮罩的方式。

默认情况下,Keras中的ImageDataGenerator可以输出文件夹的数量,无论您如何手动构建和调整ImageDataGenerator以进行图像分割而不是图像分类。

我的建议是遵循整个帖子,一开始什么都不改变。如果你注意的话,最终得到的结果是相当好的;这意味着数据集准备过程(掩码创建(是正确的。

最新更新