在tensorflow中导入图像

Question

我正在尝试导入图像并将它们转换为张量。所有其他解决方案都建议制作 filename_queue 并使用 tf.reader()，但我无法让它工作...因此，我只是使用基础知识。

我的桌面目录中有一个名为 test.jpg 的文件，我是运行一个 linux 环境。这是我的代码：

import tensorflow as tf

image = tf.image.decode_jpeg("~/Desktop/test.jpg", channels=1)
print(image)

如您所见，一些非常简单的代码...但是它会输出

Tensor("DecodeJpeg:0", shape=(?, ?, 1), dtype=uint8)

这告诉我它没有正确读取文件。我做错了什么吗？

谢谢！

Answer 1

函数tf.image.decode_jpeg只是构造一个图节点并将其添加到执行图。您实际上必须评估节点运行并将其值作为 numpy 数组。尝试以下操作：

import tensorflow as tf

image = tf.image.decode_jpeg(tf.read_file("~/Desktop/test.jpg"), channels=1)
sess = tf.InteractiveSession()
print(sess.run(image))

注意 tf.image.decode_jpeg 本身的参数应该是一个字符串 Tensor。值得庆幸的是，TensorFlow 有一个将文件读入 Tensor 的操作，即 tf.read_file.

在tensorflow中导入图像

Importing images in tensorflow

python

filesystems

image

tensorflow