image - 有谁知道为什么 input_shape 中的最后一个参数必须在 keras 的应用程序中指定为 3 ？

Question

我想使用预训练Net，例如VGG、ResNet。在 Keras 中，必须在 input_shape 的 (w,h,3) 中指定格式。如果我想将 channel 指定为1，还有更多技巧吗？

conv_vgg = keras.application.VGG16(input_shape=(224,224,3))

我想指定 3 到 1:

conv_vgg = keras.application.VGG16(input_shape=(224,224,1))

提前致谢!

Answer 1

在 imagenet 或其他图像数据集中训练的预训练网络。这意味着使用 RGB 图像进行训练，这就是为什么使用预训练网络需要三个 channel 。

如果您想对单 channel 图像使用预训练网络，您可以重复您的 channel 三次并继续。 (将 1 channel 图像再重复复制两次，从 (224,224,1) 形状到 (224,224,3) 形状(3 channel 图像)。