torchvision.datasets数据加载失败

最新推荐文章于 2024-05-16 16:12:19 发布

原创最新推荐文章于 2024-05-16 16:12:19 发布 · 505 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

本文讲述了在使用torchvision.datasets时遇到数据加载失败的情况，提供了通过自动下载CIFAR-10数据集并手动解压的方法，强调了正确设置根目录、数据类型（训练/测试）和预处理选项的重要性。

torchvision.datasets数据加载失败

如何使用torchvision.datasets进行自动下载数据失败，可以使用手动下载数据

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-c0PpwUqI-1691315259002)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20230806174124553.png)]$

Ctrl+点击可以进入相关包文件，查找下载地址：https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz

在这里插入图片描述

手动下载之后解压：

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-GXatB63t-1691315259004)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20230806174600194.png)]$

下载好数据集后，解压压缩文件，将cifar-10-batches-py拷贝到你的项目中的data文件夹内，

注：

不能修改cifar-10-batches-py文件名。
注意当前.py文件与data包的相对位置(程序中采用相对路径进行读取)。

说明下：

root : 表示CIFAR10数据的加载目录；
train : 表示是否加载数据库的训练集，False的时候加载测试集；
download : 表示是否自动下载cifar数据集；
transform : 表示是否需要对数据进行预处理，none为不进行预处理；