使用tensorflow读取小狗的图片

import tensorflow as tf
import os


def read_dog(file_list):
    # 构造文件名队列
    file_queue = tf.train.string_input_producer(file_list)
    # 构造图片阅读器
    reader = tf.WholeFileReader()
    key,value = reader.read(file_queue)
    print("key:\n", key)
    print("value:\n", value)

  # 解码
   image_decoded = tf.image.decode_jpeg(value)
   print("image_decoded:\n", image_decoded)

   # 基本的数据处理
   # 缩小图片大小
   image_resized = tf.image.resize_images(image_decoded, (200, 200))
   print("image_resized:\n", image_resized)
   image_resized.set_shape([200,200,3])

   print("image_resized:\n:", image_resized)

   # 批处理
   image_batch = tf.train.batch([image_resized], batch_size=10, num_threads=1, capacity=10)
    return image_batch

if __name__ == '__main__':
    file_name = os.listdir("C:\\Users\\Lit_leaf\\Desktop\\Machine learning\\dog")
    # print(file_name)
    file_list = [os.path.join("C:\\Users\\Lit_leaf\\Desktop\\Machine learning\\dog", file) for file in file_name]
    # print(file_list)
    image = read_dog(file_list)
    with tf.Session() as sess:
        coord = tf.train.Coordinator()
        # 创建线程
        threads = tf.train.start_queue_runners(sess=sess, coord=coord)
        image_shape = sess.run(image)
        print('图片值：\n', image_shape.shape)
        # 请求停止
        coord.request_stop()
        # 资源回收
        coord.join(threads)