将细分掩码作为细分网络的输入?

时间:2020-05-31 05:37:26

标签: keras image-segmentation cnn generative-adversarial-network semantic-segmentation

我想训练一个语义分割网络。 我有一个包含红色,蓝色,绿色,红外,低分割蒙版,高分割蒙版的数据集。

低分割蒙版由作为像素值的类标签组成,例如对于森林为0,对于水为1。尺寸为(256x256x1)。

高细分蒙版由作为像素值的类标签组成,例如森林为0,水为1。尺寸为(256x256x1)。但是根据图片标记的更精细。

我可以输入RGB IR和低分割蒙版作为语义分割网络(如UNET)的输入吗? 输出高分割蒙版。

我的数据很少,只有1000张图像(带有高分割蒙版),并且必须预测5000张图像(这些高分割蒙版对我来说是不可访问的)。这就是为什么我想尝试这种技术的原因。

我目前正在从事CGAN实施,其中Generator的损失为“ categorical_crossentropy”。 UNET架构也是如此。

0 个答案:

没有答案