将mnist数据集保存成numpy格式

最新推荐文章于 2023-09-24 05:00:00 发布

aiwanghuan5017

最新推荐文章于 2023-09-24 05:00:00 发布

阅读量1.1k

点赞数

文章标签： python 人工智能

原文链接：http://www.cnblogs.com/marsggbo/p/9883247.html

版权

本文提供了将MNIST数据集转换为NumPy格式的代码资源链接，适用于Python和人工智能领域的研究。

摘要由CSDN通过智能技术生成

import numpy as np
from urllib import request
import gzip
import pickle

filename = [
["training_images","train-images-idx3-ubyte.gz"],
["test_images","t10k-images-idx3-ubyte.gz"],
["training_labels","train-labels-idx1-ubyte.gz"],
["test_labels","t10k-labels-idx1-ubyte.gz"]
]

def download_mnist():
    base_url = "http://yann.lecun.com/exdb/mnist/"
    for name in filename:
        print("Downloading "+name[1]+"...")
        request.urlretrieve(base_url+name[1], name[1])
    print("Download complete.")

def save_mnist():
    mnist = {}
    for name in filename[:2]:
        with gzip.open(name[1], 'rb') as f:
            mnist[name[0]] = np.frombuffer(f.read(), np.uint8, offset=16).reshape(-1,28*28)
    for name in filename[-2:]:
        with g