avatar

pytorch中引入mnist数据集

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

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

路径结构为

image-20210416154219469

如果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
)
image-20210416155017835
Author: Michelle19l
Link: https://gitee.com/michelle19l/michelle19l/2021/04/16/python/pytorch/mnist文件引入/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.
Donate
  • 微信
    微信
  • 支付寶
    支付寶