1

我正在运行以下代码段:

import tensorflow as tf
import scipy.misc
from tensorflow.python.lib.io import file_io

file = file_io.FileIO('gs://BUCKET/data/celebA/000007.jpg', mode='r')
img = scipy.misc.imread(file)

如果我在 Cloud Console 中运行该代码段,我会得到一个正确的数组。但是当同样的代码片段在 Cloud ML 中运行时,img 对象是

<PIL.JpegImagePlugin.JpegImageFile image mode=RGB size=178x218 at 0x7F1F8F26DA10>

这个 stackoverflow 答案表明安装 PIL 时未安装 libjpeg。Cloud ML 运行时版本列表显示,对于Tensorflow 0.12,libjpeg-dev 是已安装的 debian 软件包。

4

1 回答 1

0

我能够在 Cloud ML 上重现此问题,这似乎是 Tensorflow 0.12.1 中 file_io 版本的问题,如果安装了 Tensorflow 1.0,它就会消失。

如果可以,请升级到 TF 的 1.0 版本。

如果您需要 0.12 版本,Cloud ML “0.12”运行时使用 TF 的官方 0.12.1 版本,但如果您愿意,可以上传自己的版本进行安装。我没有准确追踪问题何时解决,但从 2 月 2 日开始的Nightly Tensorflow 构建似乎有效。

于 2017-03-16T17:08:09.517 回答