Tensorflow 入门学习15.下载MNIST手写体数据集及保存图片
下载数据集
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import input_data
# 60000行的训练数据集(mnist.train)和10000行的测试数据集(mnist.test)
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
将数据集保存为图片
import struct
import numpy as np
import PIL.Image
filename='./MNIST_data/train-images.idx3-ubyte'
binfile=open(filename,'rb')
buf=binfile.read()
index=0
magic,numImages,numRows,numColumns=struct.unpack_from('>IIII',buf,index)
index+=struct.calcsize('>IIII')
for image in range(0,numImages):
im=struct.unpack_from('>784B',buf,index)
index+=struct.calcsize('>784B')
im=np.array(im,dtype='uint8')
im=im.reshape(28,28)
im=PIL.Image.fromarray(im)
im.save('mnist_train/train_%s.bmp'%image,'bmp')
参考: