使用转换mnist数据库保存为bmp图片

转自:http://blog.csdn.net/u010194274/article/details/50817999

转自:点击打开链接

http://yann.lecun.com/exdb/mnist/ 网上下载好数据集之后解压,得到如下四个文件。
这里写图片描述 
但是这些个文件怎么用呢,参照网站上的说明,他们都是二进制文件,格式分别如下: 
这里写图片描述
这里写图片描述 
这里写图片描述

我这里只做train_image的过程,其他几个文件类似。参照http://www.cnblogs.com/x1957/archive/2012/06/02/2531503.html

[python]  view plain  copy
  1. #! /usr/bin/python  
  2. # -*- coding: utf-8 -*-  
  3. import struct  
  4. import numpy as np  
  5. import  matplotlib.pyplot as plt  
  6. import Image  
  7. #二进制的形式读入  
  8. filename='/home/jack/Downloads/train-images.idx3-ubyte'  
  9. binfile=open(filename,'rb')  
  10. buf=binfile.read()  
  11. #大端法读入4个unsigned int32  
  12. #struct用法参见网站 http://www.cnblogs.com/gala/archive/2011/09/22/2184801.html  
  13.   
  14. index=0  
  15. magic,numImages,numRows,numColumns=struct.unpack_from('>IIII',buf,index)  
  16. index+=struct.calcsize('>IIII')  
  17. #将每张图片按照格式存储到对应位置  
  18. for image in range(0,numImages):  
  19.     im=struct.unpack_from('>784B',buf,index)  
  20.     index+=struct.calcsize('>784B')  
  21.    #这里注意 Image对象的dtype是uint8,需要转换  
  22.     im=np.array(im,dtype='uint8')  
  23.     im=im.reshape(28,28)  
  24.    # fig=plt.figure()  
  25.    # plotwindow=fig.add_subplot(111)  
  26.    # plt.imshow(im,cmap='gray')  
  27.    # plt.show()  
  28.     im=Image.fromarray(im)  
  29.     im.save('train/train_%s.bmp'%image,'bmp')  


如此就成功了 
这里写图片描述


如果提示错误:No module named Image.请这样导入import PIL.Image as Image


猜你喜欢

转载自blog.csdn.net/u012530451/article/details/78673961