学习《动手学深度学习》的过程中,下载图片分类数据集时遇到了下载失败的问题:
import torch
import torchvision # 计算机视觉相关库
from torch.utils import data
from torchvision import transforms
from d2l import torch as d2l
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)
RuntimeError: Error downloading train-images-idx3-ubyte.gz
猜测是由于国外网站下载速度过慢导致的,但开启全局代理后仍无法解决问题。
解决方法
第一步:从github下载数据集和labels,地址:https://github.com/zalandoresearch/fashion-mnist
需要下载四个文件:
- 训练数据图片
train-images-idx3-ubyte
- 训练数据标签
train-labels-idx1-ubyte
- 测试数据图片
t10k-images-idx3-ubyte
- 测试数据标签
t10k-labels-idx1-ubyte
第二步:在存放数据的文件夹中建立FashionMNIST
文件夹,包含processed
和raw
两个子文件夹,将下载好的数据放入raw
文件夹。
第三步:运行之前的代码,会跳过下载步骤直接进行解压缩,生成所需文件
之后可以按照教材顺利进行下一步操作。