MNIST数据转图片

这一步其实不重要....因为我们一般过程是将图片转为二进制数据，这是其逆过程。

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
import os
import cv2 
import numpy as np

mnist = input_data.read_data_sets("MNIST/",one_hot = True)

save_dir = 'picture/'
if os.path.exists(save_dir) is False:
    os.makedirs(save_dir)

lens = mnist.train.images.shape[0]
for i in range(lens):
    image = mnist.train.images[i,:].reshape(28,28)*255    #mnist原本的像素数据在0~1之间，作为灰度图像乘255
    file_name = save_dir+'mnist_train'+str(i)+'.png' 
    cv2.imwrite(file_name,image)