政策资讯

Policy Information


TF:利用TF读取数据操作,将CIFAR-10 数据集中的训练图片读取出来,并保存为.jpg格式

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

TF:利用TF读取数据操作,将CIFAR-10 数据集中的训练图片读取出来,并保存为.jpg 格式

目录

输出结果

核心代码


输出结果

核心代码

  1. def inputs_origin(data_dir):
  2. filenames = [os.path.join(data_dir, 'data_batch_%d.bin' % i)
  3. for i in range(1, 6)]
  4. ……
  5. filename_queue = tf.train.string_input_producer(filenames)
  6. read_input = cifar10_input.read_cifar10(filename_queue)
  7. reshaped_image = tf.cast(read_input.uint8image, tf.float32)
  8. return reshaped_image
  9. if __name__ == '__main__':
  10. with tf.Session() as sess:
  11. reshaped_image = inputs_origin('cifar10_data/cifar-10-batches-bin')
  12. threads = tf.train.start_queue_runners(sess=sess)
  13. sess.run(tf.global_variables_initializer())
  14. if not os.path.exists('cifar10_data/raw/'):
  15. os.makedirs('cifar10_data/raw/')
  16. for i in range(30):
  17. image_array = sess.run(reshaped_image)
  18. scipy.misc.toimage(image_array).save('cifar10_data/raw/%d.jpg' % i)

评论

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