Pytorch加载本地CIFAR10数据集

在线下载经常报错,可以预先下载好数据集放到本地。

  1. 下载数据集(官网页面:http://www.cs.toronto.edu/~kriz/cifar.html 下载地址:http://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz),使用IDM很快就下载完了,不要听某些人瞎忽悠去下载还要积分的资源,百度云也大可不必。
  2. 把cifar-10-python.tar.gz更名为cifar-10-batches-py.tar.gz并放到本地文件夹里,这里放到./data目录下,然后用 tar -zxvf cifar-10-batches-py.tar.gz 命令解压。
  3. 加载数据,记得设置download = False。如果上一步不知道该把数据集放到哪里,可以先设置为True,然后看下载位置在哪,之后替换掉。
import torchvision
train_dataset = torchvision.datasets.CIFAR10(root = './data/', train = True, transform = data_transforms, download = False)

完事。以上是在Linux系统里的方法,Windows没有测试,可以自行尝试。

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