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

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