政策资讯

Policy Information


TF学习——TF数据读取:TensorFlow中数据读这三张图片的5个epoch +把读取的结果重新存到read 文件夹中

来源: 重庆市软件正版化服务中心    |    时间: 2022-09-19    |    浏览量: 51157    |   

TF学习——TF数据读取:TensorFlow中数据读这三张图片的5个epoch +把读取的结果重新存到read 文件夹中

目录

实验展示

代码实现


实验展示

代码实现

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

  1. import tensorflow as tf
  2. with tf.Session() as sess:
  3. filename = ['A.jpg', 'B.jpg', 'C.jpg']
  4. filename_queue = tf.train.string_input_producer(filename, shuffle=False, num_epochs=5)
  5. reader = tf.WholeFileReader()
  6. key, value = reader.read(filename_queue)
  7. tf.local_variables_initializer().run()
  8. threads = tf.train.start_queue_runners(sess=sess)
  9. i = 0
  10. while True:
  11. i += 1
  12. image_data = sess.run(value)
  13. with open('read/test_%d.jpg' % i, 'wb') as f:
  14. f.write(image_data)

相关文章

TF之数据读取操作:TensorFlow中的数据读取,读取这三张图片的5个epoch 并且把读取的结果重新存到read 文件夹中

评论

产品推荐

更多 >

QQ咨询 扫一扫加入群聊,了解更多平台咨询
微信咨询 扫一扫加入群聊,了解更多平台咨询
意见反馈
立即提交
QQ咨询
微信咨询
意见反馈