美文网首页
pytorch如何使用本地的mnist数据集

pytorch如何使用本地的mnist数据集

作者: 锦绣拾年 | 来源:发表于2020-10-15 14:30 被阅读0次

数据集下载太慢or下载不下来。
踩坑:
https://github.com/foowaa/torchvision-datasets-mnist

从以上链接下载处理好的pt数据,
改成'training.pt'
解压processed,把它放入类名的文件夹中,如FashionMNIST新建,FashionMNIST文件夹放入processed。MNIST新建MNIST文件夹放入processed。
root写入MNIST所在目录

train_dataset = torchvision.datasets.FashionMNIST(root="\\dataset\\",
                                             train=True, 
                                             transform=transforms.ToTensor(),
                                             download=False)
test_dataset = torchvision.datasets.FashionMNIST(root="\\dataset\\", 
                                          train=False, 
                                          transform=transforms.ToTensor(),
                                          download=False)

参考源码https://github.com/pytorch/vision/blob/16ef312eed46e928ce1e2ac27490a358a72a254c/torchvision/datasets/mnist.py#L121

相关文章

网友评论

      本文标题:pytorch如何使用本地的mnist数据集

      本文链接:https://www.haomeiwen.com/subject/rlznpktx.html