如何在tf.decode_image之后显示图像的代码

Question

我正在尝试查看tf如何解码图像，所以我尝试

import tensorflow as tf
image1 = tf.image.decode_png('/usr/src/pycharm-2017.1/bin/pycharm.png')
print(image1.shape)
with tf.Session() as sess:
    img = sess.run(image1)
    print(img.shape, img)

但这会引发错误

InvalidArgumentError (see above for traceback): Invalid PNG header, data size 39
 [[Node: DecodePng = DecodePng[channels=0, dtype=DT_UINT8, _device="/job:localhost/replica:0/task:0/cpu:0"](DecodePng/contents)]]

我也尝试过tf.image.decode_image，但是它也不起作用。 怎么了 ？ 我该如何解决？ 感谢你

Answer 1

tf.image.decode_png接受字符串类型的Tensor，因此您需要先使用Tensorflow读取png，然后再将其传递给函数：

import tensorflow as tf
image1 = tf.image.decode_png(tf.read_file('/usr/src/pycharm-2017.1/bin/pycharm.png'))
print(image1.shape)
with tf.Session() as sess:
    img = sess.run(image1)
    print(img.shape, img)