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

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