生成 RGB 和灰度模式的语义分割掩码

Question

我一直在尝试为语义分割生成掩码，但我从来没有生成正确的形式。

我有 2 类分割任务，我从来没有研究过语义分割。我已经在 RGB 和灰度模式下生成了蒙版。当我使用 RGB 时，它会导致交叉熵（即

Target 77 out of bounds

）出现问题，这与类的数量及其颜色有关。此外，当我使用

(ignore_index = xx

）时，给定的目标数量不断变化。 顺便说一下，我的第一堂课用白色 (255,255,255) 表示，第二堂课用红色 (255,0,0) 表示。为了解决这个问题，我将蒙版模式从 RGB 更改为灰度。错误仍然存在，我意识到颜色也是我的班级 ID，这个错误发生是因为我只有 2 个班级。在那之后，我将我的颜色从白色和红色更改为灰度中的 1 和 2，肉眼看起来不是这样，但我检查了值。我的代码已经运行了一段时间没有任何错误。

然而，损失和其他指标（即准确性）似乎不真实。我一直在想我的 CNN 没有检测到新颜色。你可以认为我没有给出任何掩码，所以 CNN 不会产生任何掩码并且 acc.is 1。因为它们都是空的且相等的。但当然这些都是我的假设。

这是我的问题： 我的错误在哪里？ 如何正确生成掩码？

请帮助我。我感到很困惑，不知道该怎么办。提前致谢。