我有点困惑。我想分割图像使用面具,有3类。我们需要考虑2种情况。有时,掩码中的所有类都在一个通道中,但数量不同(0,1,2,3)。第二种情况是,遮罩有3个通道,每个类别在不同的通道中。
我相信下面的代码适用于第一种情况。但是我需要为第二个案子做些不同的事情吗。
对于最后一层,以下两种情况下的类数是否正确:
outputs = layers.Conv2D(num_classes, 3, activation="softmax", padding="same")(x)
还有,编译呢
model.compile(optimizer="rmsprop", loss="sparse_categorical_crossentropy")
