将mnist转换为图片以及标签转文档

本人的jupyter notebook的默认工作路径为E:/python。故将Mnist数据集也放入此文件夹下

from PIL import Image
import struct




def read_image(filename):
  f = open(filename, 'rb')




  index = 0
  buf = f.read()




  f.close()




  magic, images, rows, columns = struct.unpack_from('>IIII' , buf , index)
  index += struct.calcsize('>IIII')




  for i in range(images):
  #for i in xrange(2000):
    image = Image.new('L', (columns, rows))




    for x in range(rows):
      for y in range(columns):
        image.putpixel((y, x), int(struct.unpack_from('>B', buf, index)[0]))
        index += struct.calcsize('>B')




    print ('save ' + str(i) + 'image')
    image.save('E:/python file/test/' + str(i) + '.png')




def read_label(filename, saveFilename):
  f = open(filename, 'rb')
  index = 0
  buf = f.read()




  f.close()




  magic, labels = struct.unpack_from('>II' , buf , index)
  index += struct.calcsize('>II')
  
  labelArr = [0] * labels
  #labelArr = [0] * 2000




  for x in range(labels):
  #for x in xrange(2000):
    labelArr[x] = int(struct.unpack_from('>B', buf, index)[0])
    index += struct.calcsize('>B')




  save = open(saveFilename, 'w')




  save.write(','.join(map(lambda x: str(x), labelArr)))
  save.write('\n')




  save.close()
  print('save labels success')




if __name__ == '__main__':
#   read_image('E:/python file/train-images.idx3-ubyte')
  read_label('E:/python file/train-labels.idx1-ubyte', 'E:/python file/test/label.txt')