__init__.py 2.63 KB
Newer Older
1
from ._optical_flow import FlyingChairs, FlyingThings3D, HD1K, KittiFlow, Sintel
2
from ._stereo_matching import CarlaStereo, Kitti2012Stereo, Kitti2015Stereo
3
4
from .caltech import Caltech101, Caltech256
from .celeba import CelebA
Soumith Chintala's avatar
Soumith Chintala committed
5
from .cifar import CIFAR10, CIFAR100
6
from .cityscapes import Cityscapes
Philip Meier's avatar
Philip Meier committed
7
from .clevr import CLEVRClassification
8
from .coco import CocoCaptions, CocoDetection
puhuk's avatar
puhuk committed
9
from .country211 import Country211
Philip Meier's avatar
Philip Meier committed
10
from .dtd import DTD
11
from .eurosat import EuroSAT
12
from .fakedata import FakeData
Philip Meier's avatar
Philip Meier committed
13
from .fer2013 import FER2013
14
from .fgvc_aircraft import FGVCAircraft
15
from .flickr import Flickr30k, Flickr8k
Zhiqiang Wang's avatar
Zhiqiang Wang committed
16
from .flowers102 import Flowers102
17
from .folder import DatasetFolder, ImageFolder
Joao Gomes's avatar
Joao Gomes committed
18
from .food101 import Food101
Sumukh Aithal's avatar
Sumukh Aithal committed
19
from .gtsrb import GTSRB
20
from .hmdb51 import HMDB51
Philip Meier's avatar
Philip Meier committed
21
from .imagenet import ImageNet
22
from .inaturalist import INaturalist
23
from .kinetics import Kinetics, Kinetics400
Prabhat Roy's avatar
Prabhat Roy committed
24
from .kitti import Kitti
25
from .lfw import LFWPairs, LFWPeople
26
from .lsun import LSUN, LSUNClass
27
from .mnist import EMNIST, FashionMNIST, KMNIST, MNIST, QMNIST
28
from .omniglot import Omniglot
Philip Meier's avatar
Philip Meier committed
29
from .oxford_iiit_pet import OxfordIIITPet
30
from .pcam import PCAM
31
32
from .phototour import PhotoTour
from .places365 import Places365
33
from .rendered_sst2 import RenderedSST2
34
35
36
from .sbd import SBDataset
from .sbu import SBU
from .semeion import SEMEION
37
from .stanford_cars import StanfordCars
38
from .stl10 import STL10
Saswat Das's avatar
Saswat Das committed
39
from .sun397 import SUN397
40
41
42
43
from .svhn import SVHN
from .ucf101 import UCF101
from .usps import USPS
from .vision import VisionDataset
44
from .voc import VOCDetection, VOCSegmentation
45
from .widerface import WIDERFace
soumith's avatar
soumith committed
46

47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
__all__ = (
    "LSUN",
    "LSUNClass",
    "ImageFolder",
    "DatasetFolder",
    "FakeData",
    "CocoCaptions",
    "CocoDetection",
    "CIFAR10",
    "CIFAR100",
    "EMNIST",
    "FashionMNIST",
    "QMNIST",
    "MNIST",
    "KMNIST",
62
    "StanfordCars",
63
    "STL10",
Saswat Das's avatar
Saswat Das committed
64
    "SUN397",
65
66
67
68
69
70
71
    "SVHN",
    "PhotoTour",
    "SEMEION",
    "Omniglot",
    "SBU",
    "Flickr8k",
    "Flickr30k",
Zhiqiang Wang's avatar
Zhiqiang Wang committed
72
    "Flowers102",
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
    "VOCSegmentation",
    "VOCDetection",
    "Cityscapes",
    "ImageNet",
    "Caltech101",
    "Caltech256",
    "CelebA",
    "WIDERFace",
    "SBDataset",
    "VisionDataset",
    "USPS",
    "Kinetics400",
    "Kinetics",
    "HMDB51",
    "UCF101",
    "Places365",
    "Kitti",
    "INaturalist",
    "LFWPeople",
    "LFWPairs",
93
94
    "KittiFlow",
    "Sintel",
95
    "FlyingChairs",
96
    "FlyingThings3D",
97
    "HD1K",
Joao Gomes's avatar
Joao Gomes committed
98
    "Food101",
Philip Meier's avatar
Philip Meier committed
99
    "DTD",
Philip Meier's avatar
Philip Meier committed
100
    "FER2013",
Sumukh Aithal's avatar
Sumukh Aithal committed
101
    "GTSRB",
Philip Meier's avatar
Philip Meier committed
102
    "CLEVRClassification",
Philip Meier's avatar
Philip Meier committed
103
    "OxfordIIITPet",
104
    "PCAM",
puhuk's avatar
puhuk committed
105
    "Country211",
106
    "FGVCAircraft",
107
    "EuroSAT",
108
    "RenderedSST2",
109
110
111
    "Kitti2012Stereo",
    "Kitti2015Stereo",
    "CarlaStereo",
112
)