Python处理MINIST数据

最新推荐文章于 2023-07-13 10:47:38 发布

aitazhixin

最新推荐文章于 2023-07-13 10:47:38 发布

阅读量669

点赞数

分类专栏： AI 文章标签： python LeNet5

本文链接：https://blog.csdn.net/aitazhixin/article/details/73122479

版权

AI 专栏收录该内容

17 篇文章 0 订阅

订阅专栏

将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()