【解决】pytorch 加载本地 FashionMNIST 数据集的问题:

本文指导如何在国内环境下解决FashionMNIST数据集下载问题,提供本地下载和路径调整的方法,确保在Jupyter Notebook中成功加载数据集。

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

问题: 最近在学习沐神的动手学深度学习,第二章 linear-networks 中的图像分类数据集的时候,根据书中代码下载 FashionMNIST 数据集并导入的时候下载失败

Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-images-idx3-ubyte.gz
Failed to download (trying next):
<urlopen error [WinError 10060] 由于连接方在一段时间后没有正确答复或连接的主机没有反应,连接尝试失败。>

%matplotlib inline
import torch
import torchvision
from torch.utils import data
from torchvision import transforms
from d2l import torch as d2l

d2l.use_svg_display()
# 通过ToTensor实例将图像数据从PIL类型变换成32位浮点数格式,
# 并除以255使得所有像素的数值均在0到1之间
trans = transforms.ToTensor()
mnist_train = torchvision.datasets.FashionMNIST(
    root="../data", train=True, transform=trans, download=True)
mnist_test = torchvision.datasets.FashionMNIST(
    root="../data", train=False, transform=trans, download=True)

原因: 使用书中代码下载 FashionMNIST 数据集,默认从 amazonaws 下载,国内速度很慢或无法下载

解决办法: 提前下载好 FashionMNIST 数据集,从本地导入并加载数据集

1、如果您使用的是 Jupyter notebook,则在本节代码文件的同级目录下新建一个 data 文件夹

2、在该 data 文件夹下新建 \FashionMNIST\raw 文件夹,下载 FashionMNIST 数据集(4 个 .gz 文件),并将这四个文件加压到 \data\FashionMNIST\raw 文件夹下

FashionMNIST 数据集下载地址: 链接:https://pan.baidu.com/s/12bA68Xzef80RVs3H3XqS4A 
提取码:ui70

3、将代码中 root="../data" 改为 root="./data"

# 通过ToTensor实例将图像数据从PIL类型变换成32位浮点数格式,
# 并除以255使得所有像素的数值均在0到1之间
trans = transforms.ToTensor()    # 将图片转换为 tensor
mnist_train = torchvision.datasets.FashionMNIST(
    root="./data", 
    train=True, transform=trans, download=True)
mnist_test = torchvision.datasets.FashionMNIST(
    root="./data", 
    train=False, transform=trans, download=True)

4、重启 Jupyter 内核,重新运行 cell

得到 60000 张训练数据集和 10000 张测试数据集图片(已转换为 tensor)

解决! 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值