我想使用预训练的网络,如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))
提前感谢!
回答:
预训练网络是在ImageNet或其他图像数据集上训练的。这意味着它们是用RGB图像训练的,因此使用预训练网络需要三个通道。
如果你想对单通道图像使用预训练网络,你可以将你的通道重复三次然后继续。(将你的1通道图像复制两份,从(224,224,1)形状变为(224,224,3)形状,即3通道图像)。