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

Question

我想用pre-train Net，比如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 次并继续。（重复复制两次您的 1 通道图像，从 (224,224,1) 形状到 (224,224,3) 形状（3 通道图像）。