使用NumPy获取MNIST图片

MNIST图片

在 MNIST 数据集中的每张图片由 28 * 28 个像素点构成，每个像素点用一个灰度值表示。在这里，我们将 28 * 28 的像素展开并将其写入到指定目录。

首先，我们建立展开后的目录，我的配置如下图。也就是在minst_data（道歉，才发现目录写错）下建立 mnist_train 目录。

Python代码

import struct
import numpy as np
#import matplotlib.pyplot as plt
import PIL.Image

filename='minst_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('minst_data/mnist_train/train_%s.bmp'%image,'bmp')

注意目录的位置要和自己机器上配置一致。

运行结果

代码运行后，可以在 mnist_train 目录下得到对应的原始图片。如下图所示。图片比较多，耐心等。