语义分割数据集组织
Semantic segmentation dataset organization
我正在尝试使用语义分割来分割 4 个病灶。我跟着这个
this great post
我的训练文件夹只有 2 个带有补丁的子文件夹：masks 和 images。在带掩码的文件夹内，所有 类 都混合在一起。另一个文件夹有相应的图像。因此，当我训练模型时，它出现：ONE CLASS FOUND，就在上述 post 之后。结果令人失望，我想知道是否必须在文件夹中拆分 类，因此模型识别 4 类 而不是一个。
您真正需要注意的是创建蒙版的方式。
无论您如何手动构建和调整 ImageDataGenerator 用于图像分割而不是图像分类，Keras 中的 ImageDataGenerator 都可能默认输出文件夹数。
我的建议是遵循整个 post，一开始不要做任何更改。如果你注意的话，最后得到的结果是相当不错的；这意味着数据集准备过程（掩码创建）是正确的。
我正在尝试使用语义分割来分割 4 个病灶。我跟着这个 this great post
我的训练文件夹只有 2 个带有补丁的子文件夹：masks 和 images。在带掩码的文件夹内，所有 类 都混合在一起。另一个文件夹有相应的图像。因此，当我训练模型时，它出现：ONE CLASS FOUND，就在上述 post 之后。结果令人失望，我想知道是否必须在文件夹中拆分 类，因此模型识别 4 类 而不是一个。
您真正需要注意的是创建蒙版的方式。
无论您如何手动构建和调整 ImageDataGenerator 用于图像分割而不是图像分类，Keras 中的 ImageDataGenerator 都可能默认输出文件夹数。
我的建议是遵循整个 post，一开始不要做任何更改。如果你注意的话，最后得到的结果是相当不错的；这意味着数据集准备过程（掩码创建）是正确的。