深度学习-如何下载以及导入Datasets数据集

本文介绍了如何下载和导入深度学习中常用的Datasets数据集。提供了在线和离线两种下载方式,强调了离线下载时的注意事项,特别是指出了在设置本地路径时,需要确保路径末尾添加斜杠以避免运行时错误。

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

1.在线下载

可以使用以下代码在线下载Datasets数据集

mnist_train=torchvision.datasets.FashionMNIST(root='~/Datasets/FashionMNIST',train=True,download=True,transform=transforms.ToTensor())
mnist_test=torchvision
PyTorch 中可以使用 `torchvision` 库来离线导入 MNIST 数据集。MNIST 是一个手写数字识别的经典数据集,它包含了训练集和测试集,每张图片都是 28x28 灰度图像,标签为0到9的数字。 以下是导入 MNIST 数据集的基本步骤: 1. 首先,确保已经安装了 PyTorch 和 torchvision 库,如果没有,可以通过 pip 安装: ```bash pip install torch torchvision ``` 2. 导入所需的库: ```python import torch from torchvision import datasets, transforms ``` 3. 设置数据加载器的选项,包括图像大小、是否归一化等,并指定数据集下载的位置(如果不存在的话): ```python transform = transforms.Compose([ transforms.ToTensor(), # 转换成 PyTorch tensor transforms.Normalize((0.5,), (0.5,)) # 归一化,平均值和标准差通常设为 0.5 和 0.5 ]) # 如果是本地加载,设置成False train_data = datasets.MNIST(root='./data', train=True, download=True, transform=transform) test_data = datasets.MNIST(root='./data', train=False, download=True, transform=transform) ``` 这里 `root='./data'` 表示数据将存储在当前目录下的 'data' 文件夹下。 4. 加载数据: ```python train_loader = torch.utils.data.DataLoader(train_data, batch_size=64, shuffle=True) # 创建训练数据加载器 test_loader = torch.utils.data.DataLoader(test_data, batch_size=64, shuffle=False) # 创建测试数据加载器 ``` 现在你有了训练集和测试集的数据加载器,可以直接用于模型训练和评估。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值