如何在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)