首页 文章

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

提问于
浏览 1984
1

我想使用前列车网,比如VGG,ResNet . 在Keras中,必须在input_shape中指定(w,h,3)中的格式 . 如果我想将 Channels 指定为1,是否有更多技巧?

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

我想指定3比1:

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

提前致谢!

1 回答

  • 1

    在imagenet或其他图像数据集中训练的预训练网络 . 这意味着使用RGB图像进行训练,这就是使用预训练网络需要三个通道的原因 .

    如果您想为单个 Channels 图像使用预先训练的网络,您可以重复三次 Channels 并继续 . (重复复制两次1通道图像,从(224,224,1)形状到(224,224,3)形状(3通道图像) .

相关问题