torchvision中数据集的使用（Pycharm）

前言：

_我们在学习深度学习基础的时候，经常会因为想测试自己写的对不对，但是找不到数据集而感到烦恼，其实，Pytorch为我们提供了一些基础的数据集（当然在以后模型训练中还是要自己从网上找数据集的）；

_

官方数据集的下载：

官方网址：
PyTorch

第一种手动下载：

① 进入以后选择Docs中的torchvision

② 进入以后向下拉，会找到一个dataset

③ 点击dataset（红框中的都是数据集）:

④ 以小土堆中的CIFAR10 为例:

⑤ 点击第二个箭头所指的 CIFAR10 Dataset

⑥点击红框汇中的蓝色字体就可以进行下载。

第二种在Pycharm中使用代码进行下载：

输入以下代码

import torchvision

train_set = torchvision.datasets.CIFAR10(root="./dataset", train=True, download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset", train=False, download=True)