pytorch中引入mnist数据集时文件的放置位置，解决`runtimeerror dataset not found`

最新推荐文章于 2024-08-14 13:37:04 发布

小仙女不掉发

最新推荐文章于 2024-08-14 13:37:04 发布

阅读量3.2k

点赞数 5

分类专栏：机器学习文章标签： pytorch

本文链接：https://blog.csdn.net/mich99/article/details/115764524

版权

机器学习专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文详细介绍了如何获取和使用FashionMNIST数据集。该数据集可以从 torchvision.datasets.FashionMNIST 直接下载，或者手动下载并放置到指定的'FashionMNIST'文件夹内。代码会自动检测数据是否存在，如果需要下载，设置download=True；已有数据则设置download=False。同时，文章提醒注意数据集的文件结构，确保正确加载。

摘要由CSDN通过智能技术生成

关于数据集FashionMNist，可以选择在代码中下载，也可以手动下载https://gitee.com/mirrors/Fashion-MNIST/

mnist_train = torchvision.datasets.FashionMNIST(
              root='data/fashion/',
              train=True, download=True, transform=torchvision.transforms.ToTensor())

路径结构为
在这里插入图片描述

如果download=True，则等待下载即可。如果手动下载，注意在变量root路径下建立FashionMNIST文件夹（和torchvision.datasets.FashioMNIST中的FashionMNIST相同），将下载内容放入raw中（解压不解压均可）

注意，如果没有processed文件夹中的内容，依然需要设置download=True，建议直接设置为download=True，代码将自动检测是否需要下载数据集

如果设置不好将会报错：RuntimeError: Dataset not found.

当然也可以只放入processed文件夹，这时download=False即可

train_data_ = torchvision.datasets.MNIST( # 在root下创建文件夹"MNIST"
    root='./mnist',
    train=True,
    transform=torchvision.transforms.ToTensor(),  # np array(pixel)修改为tensor （0，255）->（0，1）
    download=False
)