zl程序教程

您现在的位置是:首页 >  工具

当前栏目

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

学习数据 图片 读取 文件夹 结果 Tensorflow 重新
2023-09-14 09:04:51 时间

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 文件夹中