Tensorflow读取mnist图像数据并保存图像

Halcom

Tensorflow读取mnist图像数据并保存图像
百度网盘链接：
链接：http://pan.baidu.com/s/1i4BbOlJ
具体链接在halcom.cn论坛，联系人QQ：3283892722
该论坛是一个学习交流平台，我会逐一的和大家分享学习。
欢迎大家录制视频，你可在论坛进行打赏分享。
视频专用播放器：http://halcom.cn/forum.php?mod=viewthread&tid=258&extra=page%3D1

1. import tensorflow as tf
   
2. import os
   
3. from PIL import Image
   
4. 
   
5. cur_dir = os.getcwd()
   
6. print("For Tensorflow to load images")
   
7. print("current directory:",cur_dir)
   
8. 
   
9. def read_image(filename_queue):
   
10.     reader = tf.WholeFileReader()
    
11.     key,value = reader.read(filename_queue)
    
12.     image = tf.image.decode_png(value)
    
13.     return key,image
    
14. 
    
15. def inputs():
    
16.     # 手写数字--灰度图像
    
17.     #filenames = ['./Test/test/1.png', './Test/test/2.png' ]
    
18.    
    
19.     FullImageList = [];
    
20.     filenames = os.listdir('./Test/test/')
    
21.     for fn in filenames:
    
22.         FullImage = os.path.join( cur_dir, 'Test/test', fn )
    
23.         FullImageList.append(FullImage)
    
24. 
    
25.     filename_queue = tf.train.string_input_producer(FullImageList)
    
26.     filename,read_input = read_image(filename_queue)
    
27.     return filename,read_input
    
28. 
    
29. with tf.Graph().as_default():
    
30.     image = inputs()
    
31.     init = tf.global_variables_initializer()
    
32.     sess = tf.Session()
    
33.     sess.run(init)
    
34.     tf.train.start_queue_runners(sess=sess)
    
35.     for i in range(3):
    
36.         filename,img = sess.run(image)
    
37.         print (filename)
    
38.         # 手写数字--灰度图像
    
39.         img = Image.fromarray(img[:,:,0], "L")
    
40.         img.save(os.path.join(cur_dir+'/Test/',"ysw"+str(i)+".bmp"))
    
41.         

复制代码