我正在研究一个有 6 个类的多类语义分割问题。我在 keras ImageDataGenerator 中增强了图像和蒙版。现在我想将我的面具转换为分类。这样我就可以将其输入到 Unet。如果我在增加掩码后尝试转换为分类，内存就会消失。如果我尝试在增强之前对其进行转换，图像生成器会抛出一个错误，因为它只支持 4 个类。有没有替代方法？
我正在研究一个有 6 个类的多类语义分割问题。我在 keras ImageDataGenerator 中增强了图像和蒙版。现在我想将我的面具转换为分类。这样我就可以将其输入到 Unet。如果我在增加掩码后尝试转换为分类，内存就会消失。如果我尝试在增强之前对其进行转换，图像生成器会抛出一个错误，因为它只支持 4 个类。有没有替代方法？