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

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

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

路径结构为
pytorch中引入mnist数据集时文件的放置位置,解决`runtimeerror dataset not found`_第1张图片

如果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
)

pytorch中引入mnist数据集时文件的放置位置,解决`runtimeerror dataset not found`_第2张图片

你可能感兴趣的:(机器学习,pytorch)