Python 3.x 在Unet中获取超过3个类的输出时出现问题

我已经尝试了上面的代码I，它可以运行3个类，但是如果我把类的数量改为5，我就不能得到正确的输出

conv9 = Conv2D(2, 3, activation = 'relu', padding = 'same', kernel_initializer = he_normal')(conv9)
conv10 = Conv2D(3, 1, activation = 'softmax')(conv9)

---

那么你得到了什么样的产出？你能提供更多的上下文和代码吗？我已经提到了一些编码部分，请检查你的第二个最后一层只有2个过滤器，因此你的最后一层试图从2个过滤器中创建5个过滤器。在第二层和最后一层中没有那么多信息。有两种可能，增加最后一层（可能是之前的一层）以拥有更多的过滤器，和/或将内核大小从1增加到3、5甚至7。告诉我这是怎么回事

conv9 = Conv2D(2, 3, activation = 'relu', padding = 'same', kernel_initializer = he_normal')(conv9)
conv10 = Conv2D(5, 1, activation = 'softmax')(conv9)