torchvision 使用本地 FashionMNIST 数据集进行导入的方法

torchvision 使用本地 FashionMNIST 数据集进行导入的方法

一、 错误示例：

不解压、随便一个路径

二、正确操作

解压它们并放在指定的地方：

你需要放在对应数据的 \FashionMNIST\raw\ 文件夹下。

具体的结构请注意是这个样子：

我们在 root 中需要指定 data 文件夹。

import torch
import numpy as np
import torchvision
from torchvision import transforms
import matplotlib.pyplot as plt

trans = transforms.ToTensor()
mnist_train = torchvision.datasets.FashionMNIST(
    root=r"data",
    train=True,
    transform=trans,
    download=False,
)
mnist_test = torchvision.datasets.FashionMNIST(
    root=r"data",
    train=False,
    transform=trans,
    download=False,
)