pytorch数据集相关模块

发布时间:2022-08-19 12:32

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 下载数据集并将其放在根目录中。如果数据集已经下载,则不会再次下载。

ItVuer - 免责声明 - 关于我们 - 联系我们

本网站信息来源于互联网,如有侵权请联系:561261067@qq.com

桂ICP备16001015号