tensorflow加载图像

具体代码如下：

import matplotlib.pyplot as plt
import tensorflow as tf

image_raw_data_jpg = tf.gfile.FastGFile("E:\\doc\\test.jpg", "rb").read()

with tf.Session() as sess:
    img_data = tf.image.decode_jpeg(image_raw_data_jpg)

    plt.figure(1)
    print(sess.run(img_data))
    plt.imshow(img_data.eval())

# 队列式读取

path = "E:\\doc\\test.jpg"
file_queue = tf.train.string_input_producer([path]) # 创建输入队列
image_reader = tf.WholeFileReader()
_, image = image_reader.read(file_queue)
image = tf.image.decode_jpeg(image)

with tf.Session() as sess:
    coord = tf.train.Coordinator() # 协同启动的线程
    threads = tf.train.start_queue_runners(sess=sess, coord=coord)
    print(sess.run(image))
    coord.request_stop() # 停止所有线程
    coord.join(threads)