在预先训练的inception3模型中尝试访问pool3激活时,如何利用批处理

时间:2016-06-14 19:45:40

标签: tensorflow

我目前可以执行以下操作:

 pool3 = sess.graph.get_tensor_by_name('pool_3:0')
 pool3_features = sess.run(pool3,{'DecodeJpeg/contents:0': data})

其中data是单个图像的二进制流

data = gfile.FastGFile(img_name, 'rb').read()

我可以将它们叠加起来,而不是单独传递每个图像流吗?如果图像都是不同的尺寸怎么办?操作图如何处理不同大小的图像?

1 个答案:

答案 0 :(得分:1)

DecodeJpeg一次只能加载一个图像。您可以看到一些将resize_bilinear操作应用到图像后的示例:

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py#L621

将多个图像调整为匹配尺寸后,可以使用expand_dims将它们展开为4D,然后沿批处理维度将它们连接在一起以获得批处理。