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

最新推荐文章于 2019-08-08 15:40:21 发布

睿智der小贝比

最新推荐文章于 2019-08-08 15:40:21 发布

阅读量163

点赞数

本文介绍了一个Python脚本，用于读取MNIST数据集的图像和标签文件，并将图像转换为PNG格式，同时将标签数据保存到文本文件中。该脚本使用了PIL库来处理图像，以及struct库来解析二进制数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本人的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')