数据

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

内置数据集

所有数据集都是torch.utils.data.Dataset即，他们已经实施了 AND 方法。 因此，它们都可以传递给__getitem____len__torch.utils.data.DataLoader它可以使用 worker 并行加载多个样本。 例如：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

图像分类

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

图像检测或分割

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

光流

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

图像对

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

图像字幕

CocoCaptions（根、annFile、转换、...

MS Coco 字幕数据。

视频分类

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

自定义数据集的基类

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