将图像Feed重新整形为张量流

Question

我在图像分类问题中使用张量流，但我感觉在与重塑输入相关的部分中丢失了，在这种情况下是图像

我使用misc拍摄照片并调整大小

image = misc.imread("actor.jpg")
resize_image = misc.imresize(image,[224, 224], interp='nearest') 

和图像形状是

  

（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)'

？的含义以及如何正确调整大小

感谢提前

Answer 1

许多图像功能都希望批次包含多个图像。第一个维度标识批次中的图像索引。如果您只有一个要处理的图像，可以使用以下代码对其进行整形：

resize_image = tf.reshape(image, [-1, 224, 224, 3])