Policy Information
TF学习——TF数据读取:TensorFlow中数据读这三张图片的5个epoch +把读取的结果重新存到read 文件夹中
目录
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)
-
相关文章
TF之数据读取操作:TensorFlow中的数据读取,读取这三张图片的5个epoch 并且把读取的结果重新存到read 文件夹中
评论