有人知道为什么在keras的应用程序中input_shape中的最后一个arg必须指定为3吗？

Question

我想使用预训练网络，例如VGG，ResNet。在Keras中时，必须在input_shape的（w，h，3）中指定格式。如果要将通道指定为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图像进行训练，这就是使用预训练网络需要三个通道的原因。

如果您想对单个频道图像使用经过预先训练的网络，则可以重复频道3次并继续。 （从（224,224,1）形状到（224,224,3）形状（3通道图像）重复复制1通道图像两次。