MNIST数据集下载(自动下载)

news2025/1/24 14:06:32

📚**MNIST数据集下载(自动下载)**📚

在这里插入图片描述

🌈 个人主页：高斯小哥
🔥 高质量专栏：Matplotlib之旅：零基础精通数据可视化、Python基础【高质量合集】、PyTorch零基础入门教程
👈 希望得到您的订阅和支持~
💡 创作高质量博文(平均质量分92+)，分享更多关于深度学习、PyTorch、Python领域的优质内容！（希望得到您的关注~）

🌵文章目录🌵

💡一、MNIST数据集简介
📥二、PyTorch中的MNIST数据集
🔧三、下载MNIST数据集
🤝四、期待和你共同进步

在深度学习的旅程中，选择一个适合的数据集是至关重要的第一步。MNIST数据集作为经典的手写数字识别数据集，为初学者和专家提供了一个简单但有效的实践平台。在本文中，我们将探讨如何利用PyTorch框架自动下载并加载MNIST数据集，以便开始我们的深度学习之旅。🚀

关键词： #MNIST数据集 #PyTorch下载 #深度学习入门 #手写数字识别

💡一、MNIST数据集简介

MNIST数据集包含60000个训练样本和10000个测试样本，每个样本都是28x28像素的手写数字灰度图像。数据集还包括每个图像对应的标签，即每个数字的实际值(0-9)。📚

📥二、PyTorch中的MNIST数据集

PyTorch提供了torchvision库，其中包含了一些常用的数据集和模型。MNIST数据集就是其中之一。我们可以使用torchvision.datasets.MNIST类轻松下载和加载MNIST数据集。📥

🔧三、下载MNIST数据集

在下载MNIST数据集之前，请确保已经安装了torch和torchvision库。如果尚未安装，可以使用以下命令进行安装：

pip install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/

接下来，我们可以使用以下代码来下载MNIST数据集：

import torch
from torchvision import datasets, transforms

# 设置数据转换，将Tensor正规化到[-1,1]之间，并添加通道维度
transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.5,), (0.5,))
])

# 下载训练集
trainset = datasets.MNIST('~/.pytorch/MNIST_data/', download=True, train=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)

# 下载测试集
testset = datasets.MNIST('~/.pytorch/MNIST_data/', download=True, train=False, transform=transform)
testloader = torch.utils.data.DataLoader(testset, batch_size=64, shuffle=True)

在这段代码中，我们首先导入了必要的库，并定义了一个数据转换transform。这个转换包括两个步骤：将图像转换为Tensor，并将像素值从[0,1]正规化到[-1,1]。这样做有助于模型训练。

然后，我们使用datasets.MNIST类下载训练集和测试集。'~/.pytorch/MNIST_data/'是数据集下载和存储的目录，download=True表示如果数据集尚未下载，则进行下载。train=True表示下载训练集，train=False表示下载测试集。transform参数用于应用之前定义的数据转换。

最后，我们使用torch.utils.data.DataLoader创建数据加载器。这个加载器可以方便地从数据集中批量加载数据，并在每个epoch中随机打乱数据。