torchvision 中数据集的使用
数据集和transfroms结合在一起的内容
标准数据集的使用
torchvision.datasets 官方指导文档:https://pytorch.org/vision/stable/datasets.html
torchvision中的数据集
关于torchvision中的datasets,有多种类型标准数据集供选用,但每个数据集中参数需要说明
例子:
CIFAR10数据集
参数
- root ( string ) – 数据集的根目录,
cifar-10-batches-py
如果下载设置为 True,则该目录存在或将保存到该目录。
- train ( bool , optional ) – 如果为真,则从训练集创建数据集,否则从测试集创建。
- transform ( callable , optional ) – 一个函数/转换,它接收 PIL 图像并返回转换后的版本。例如,
transforms.RandomCrop
- target_transform ( callable , optional ) – 接收目标并对其进行转换的函数/转换。
- download ( bool , optional ) – 如果为 true,则从 Internet 下载数据集并将其放在根目录中。如果数据集已经下载,则不会再次下载。