Pytorc读取本地的FashionMNIST数据集并加载

最新推荐文章于 2025-10-28 09:41:24 发布

原创最新推荐文章于 2025-10-28 09:41:24 发布 · 1.2k 阅读

7 ·

CC 4.0 BY-SA版权

博客介绍了FashionMNIST数据集的下载与加载方法。数据集一部分从百度云下载.pt文件并改名，另一部分从github下载。需在目录下建FashionMNIST文件夹，其下再建raw和processed子文件夹分别存放.gz和.pt文件，最后用代码加载。

下载数据集：https://github.com/foowaa/torchvision-datasets-mnist

一部分是从百度云链接下载的.pt文件，并将'train.pt'改成'training.pt'

另一部分是从github下载的文件

在你的目录下建立文件夹FashionMNIST，并在FashionMNIST文件夹下新建两个子文件夹

其中raw存放四个.gz文件processed存放两个.pt文件

之后使用代码进行加载

mnist_train = torchvision.datasets.FashionMNIST(root=r'C:/Users/xxx/datasets/',
                    train=True, transform=transforms.ToTensor(), download=False)
mnist_test = torchvision.datasets.FashionMNIST(root=r'C:/Users/xxx/datasets/',
                    train=False, transform=transforms.ToTensor(), download=False)

datasets里面包含FashionMNIST文件夹