当我想使用预训练的VGG16时，问题是它期望的形状=（None，224，224，3），但找到的形状=（32，28，28）。我该怎么做才能使用该模型？或者我不应该对244 x 244像素以下的图像使用convnet？谢谢
wydwbb8l1#
使用tensorflow 调整大小，如下所示：
(x_train, y_train), (_, _) = tf.keras.datasets.mnist.load_data() print(x_trian.shape) # (60000, 28, 28) # train set / data x_train = np.expand_dims(x_train, axis=-1) x_train = tf.image.resize(x_train, [224,224]) print(x_train.shape) # (60000, 224, 224, 1)
1条答案
按热度按时间wydwbb8l1#
使用tensorflow 调整大小，如下所示：