语义分割数据集组织

Question

我正在尝试通过语义分割对4个病变进行分割。我遵循这个 this great post

我的训练文件夹只有2个带有补丁的子文件夹：蒙版和图像。在带遮罩的文件夹中，混合了所有类。另一个文件夹具有相应的图像。因此，当我训练模型时，它出现在：刚好在上述文章之后。结果令人失望，我想知道是否必须拆分文件夹中的类，因此模型可以识别4个类，而不是一个。

Answer 1

您真正需要注意的是蒙版的创建方式。

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

我的建议是跟踪整个帖子，一开始不做任何更改。如果您注意的话，最终得到的结果将是相当不错的。这意味着数据集准备过程（掩码创建）是正确的。