重要!!!
重要!!!
重要!!!
先下载MNIST数据集,将train和test集图片解压!否则程序找不到相应文件!
注意路径!!!!!!!!!!!!!!!!!
注意路径!!!!!!!!!!!!!!!!!
注意路径!!!!!!!!!!!!!!!!!
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
import struct
from PIL import Image
import scipy.misc
filename = r'D:\source\technology_source\data\t10k-images.idx3-ubyte'
binfile = open('C:/Users/Administrator/Desktop/python/MNIST_data/train-images.idx3-ubyte', 'rb')
buf = binfile.read()
index = 0
magic, numImages, numRows, numColumns = struct.unpack_from('>IIII', buf, index) # 读取前4个字节的内容
index += struct.calcsize('>IIII')
for i in range(numImages):
im = struct.unpack_from('>784B', buf, index) # 以大端方式读取一张图上28*28=784
index += struct.calcsize('>784B')
im = np.array(im)
im = im.reshape(28, 28)
# fig = plt.figure()
# plt.axis('off')
# plt.imshow(im, cmap='gray')
# plt.show()
scipy.misc.imsave('C:/Users/Administrator/Desktop/python/train/'+"train" + str(i) + ".jpg", im)
# im = Image.fromarray(im)
# im.save("test" + str(i) + ".tif")
# plt.images.imsave("test" + str(i) + ".png", im)
plt.close()
binfile.close()