如何将图像重塑为特定尺寸

Question

我正在尝试将图像发送到我的模型，但是图像的形状与模型需要的形状不同。

ValueError                                Traceback (most recent call 
last)
<ipython-input-5-d3bd0e2a98e0> in <module>()
257 
258 else:
--> 259     model.fit({'input': X_train}, {'targets': y_train}, 
n_epoch=10,snapshot_step=500)
260     model.save('modelvgg.tfl')
261 
ValueError: Cannot feed value of shape (64, 224, 224) for Tensor ' 
input/X:0', which has shape '(?, 224, 224, 3)'

所有我想知道如何适合这些尺寸，但是我不知道如何。

Answer 1

您缺少输入中的最后一个维度，这是通道数。该模型需要3个通道-最有可能用于RGB。您正在馈送的图像只有一个通道，很可能是灰度图像。如果您根本没有RGB图像，请尝试使用np.repeat将通道尺寸复制3次。