实验展示

代码实现

1、如果设置shuffle为True ，那么在每个epoch 内图像会被打乱。该参数代表在一个epoch 内文件的顺序是否被打乱

import tensorflow as tf

with tf.Session() as sess:  

   filename = ['A.jpg', 'B.jpg', 'C.jpg']

   filename_queue = tf.train.string_input_producer(filename, shuffle=False, num_epochs=5)

   reader = tf.WholeFileReader()

   key, value = reader.read(filename_queue)

   

   tf.local_variables_initializer().run()

   threads = tf.train.start_queue_runners(sess=sess)  

   i = 0

   while True:

       i += 1

       image_data = sess.run(value)  

       with open('read/test_%d.jpg' % i, 'wb') as f:

           f.write(image_data)