将图像馈送重塑为张量流

ahm*_*ama 4 python numpy tensorflow

我在图像分类问题中使用了tensorflow,但我在与重塑输入有关的部分感到迷茫,在这种情况下,输入是重塑的

我使用杂项拍摄图像并调整其大小

image = misc.imread("actor.jpg")
resize_image = misc.imresize(image,[224, 224], interp='nearest') 
Run Code Online (Sandbox Code Playgroud)

并且图像形状是

(224,224,3)

我收到与图像不兼容有关的错误

ValueError: Cannot feed value of shape (224, 224, 3) for Tensor u'input_image_2:0', which has shape '(?, 224, 224, 3)'
Run Code Online (Sandbox Code Playgroud)

什么意思以及如何正确调整大小

提前致谢

Mat*_*ino 5

许多图像功能期望批处理包含多个图像。第一维标识批次中图像的索引。如果只有一个图像要处理,则可以使用以下代码对其进行重塑:

resize_image = tf.reshape(image, [-1, 224, 224, 3])
Run Code Online (Sandbox Code Playgroud)