Python处理MINIST数据

将MINIST图像数据处理成作为LeNet5模型输入数据的结构：

def __imprtImage(filename):filehandle = open(filename, 'rb')MSB = unpack(">I", filehandle.read(4))[0]imageNum = unpack(">I", filehandle.read(4))[0]imageRow = unpack(">I", filehandle.read(4))[0]imageCol = unpack(">I", filehandle.read(4))[0]imageData = filehandle.read()filehandle.close()imageDataArray = np.zeros((1, imageNum*32*32), np.uint8)padfile = open(filename+"pad", 'wb')for idx in range(imageNum) :imageElement = imageData[(imageRow*imageCol*idx) : (imageRow*imageCol*(idx+1))]for mdx in range(28):imageDataArray[0,(idx * 32*32 + (mdx+2)*32 + 2):(idx * 32*32 + (mdx+2)*32 + 30)] = list(imageElement[(mdx * imageCol): (mdx*imageCol+imageCol)])blist = bytes(list(imageDataArray[0,idx*32*32:(idx+1)*32*32]))padfile.write(blist)padfile.close()