MNIST数据集转换为.csv格式

PeppaPeppaPeppa

于 2019-03-27 09:54:13 发布

阅读量1.1k

点赞数

本文介绍了一个用于将MNIST数据集从原始格式转换为CSV文件的Python脚本。该脚本可以读取训练和测试集的图像及标签，并将它们转换为适合机器学习任务的格式。

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

def convert(imgf, labelf, outf, n):
    f = open(imgf, "rb")
    o = open(outf, "w")
    l = open(labelf, "rb")

    f.read(16)
    l.read(8)
    images = []

    for i in range(n):
        image = [ord(l.read(1))]
        for j in range(28*28):
            image.append(ord(f.read(1)))
        images.append(image)

    for image in images:
        o.write(",".join(str(pix) for pix in image)+"\n")
    f.close()
    o.close()
    l.close()

convert("MNIST/train-images.idx3-ubyte", "MNIST/train-labels.idx1-ubyte",
        "MNIST/mnist_train.csv", 60000)
convert("MNIST/t10k-images.idx3-ubyte", "MNIST/t10k-labels.idx1-ubyte",
        "MNIST/mnist_test.csv", 10000)
print("Convert Finished!")