数据

Torchvision 在模块中提供了许多内置数据集，以及用于构建自己的数据集的实用程序类。torchvision.datasets

内置数据集

所有数据集都是 i.即，它们具有 和 实现的方法。 因此，它们都可以传递给 a，后者可以使用 worker 并行加载多个样本。 例如：__getitem____len__torch.multiprocessing

imagenet_data = torchvision.datasets.ImageNet('path/to/imagenet_root/')
data_loader = torch.utils.data.DataLoader(imagenet_data,
                                          batch_size=4,
                                          shuffle=True,
                                          num_workers=args.nThreads)

所有数据集都有几乎相似的 API。它们都有两个常见的参数：分别转换 input 和 target。 您还可以使用提供的基类创建自己的数据集。transformtarget_transform

图像分类

（根， target_type， str] =， ...）	加州理工学院 101数据。
（根、转换、...	加州理工学院 256数据。
（root， split， target_type， str] =， ...）	大规模 CelebFaces 属性 （CelebA） 数据集数据。
（根、训练、转换......	CIFAR10数据。
（根、训练、转换......	CIFAR100数据。
（根、拆分、转换......	来自 OpenAI 的 Country211 数据集。
（根、拆分、分区、转换等）	可描述纹理数据集 （DTD）。
（根，拆分，**kwargs）	EMNIST数据。
（根、转换、target_transform、...	EuroSAT 数据集的 RGB 版本。
（大小、image_size、int、int] =， ...）	一个虚假数据集，返回随机生成的图像并将其作为 PIL 图像返回
（根、训练、转换......	时尚 MNIST数据。
（根、拆分、变换target_transform）	FER2013数据。
（根、拆分、annotation_level、...	FGVC 飞机数据。
（根、ann_file、转换......	Flickr8k 实体数据。
（根、ann_file、转换......	Flickr30k 实体数据。
（根、拆分、转换......	牛津 102 花数据。
（根、拆分、转换......	Food-101 数据集。
（根、拆分、转换......	德国交通标志识别基准 （GTSRB）数据。
（根、版本、target_type、...	iNaturalist 智能博物学家数据。
（根，拆分，**kwargs）	ImageNet 2012 分类数据集。
（根、训练、转换......	Kuzushiji-MNIST数据。
（根、拆分、image_set、转换......	LFW数据。
（根， 类， List[str]] =， 转换， ...）	LSUN 数据集。
（根、训练、转换......	MNIST数据。
（根、背景、转换等）	Omniglot （全能）数据。
（根、拆分、target_types、...	Oxford-IIIT 宠物数据集。
（root、split、small、download 等）	Places365 分类数据集。
（根、拆分、转换......	PCAM 数据集。
（root， what， compat， train， **kwargs）	QMNIST 公司数据。
（根、拆分、转换......	渲染的 SST2 数据集。
（根、转换、target_transform、...	赛美安数据。
（根、转换、target_transform、下载）	SBU 字幕照片数据。
（根、拆分、转换......	斯坦福汽车数据
（根、拆分、折叠、转换......	STL10 系列数据。
（根、转换、target_transform、...	SUN397 数据集。
（根、拆分、转换......	SVHN 系列数据。
（根、训练、转换......	美国 邮政数据。

图像检测或分割

（根、annFile、转换、...	MS Coco 检测数据。
（root， split， target_type， str] =， ...）	大规模 CelebFaces 属性 （CelebA） 数据集数据。
（root、split、mode、target_type ...	城市景观数据。
（根、拆分、转换......	德国交通标志识别基准 （GTSRB）数据。
（根、训练、转换......	KITTI数据。
（根、拆分、target_types、...	Oxford-IIIT 宠物数据集。
（root、image_set、mode、download 等）	语义边界数据集
（根、年份、image_set、...	帕斯卡 VOCSegmentation 数据集。
（根、年份、image_set、...	帕斯卡 VOC检测数据集。
（根、拆分、转换......	WIDERFace数据。

光流

（根[、拆分、转换]）	FlyingChairs 飞椅光流数据集。
（根[， 拆分， pass_name， ...]）	FlyingThings3D 数据集。
（根[、拆分、转换]）	HD1K 数据集。
（根[、拆分、转换]）	光流的 KITTI 数据集 （2015）。
（根[、拆分、pass_name、转换]）	辛特光流数据集。

图像对

（根、拆分、image_set、转换......	LFW数据。
（根、名称、训练、转换、下载）	多视图立体对应数据。

图像字幕

（根、annFile、转换、...

MS Coco 字幕数据。

视频分类

（根、annotation_path、...	HMDB51 数据集。
（根、frames_per_clip、num_classes、...	Generic Kinetics 数据集。
（根、frames_per_clip、...	Kinetics-400 数据集。
（根、annotation_path、...	UCF101 数据集。

自定义数据集的基类

（root， loader， Any]， ...）	通用数据加载器。
（根、转换、...	默认情况下，图像以这种方式排列的通用数据加载器：.
（根、转换、转换、...	Base Class 用于制作与 torchvision 兼容的数据集。