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

57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
__all__ = (
    "LSUN",
    "LSUNClass",
    "ImageFolder",
    "DatasetFolder",
    "FakeData",
    "CocoCaptions",
    "CocoDetection",
    "CIFAR10",
    "CIFAR100",
    "EMNIST",
    "FashionMNIST",
    "QMNIST",
    "MNIST",
    "KMNIST",
72
    "StanfordCars",
73
    "STL10",
Saswat Das's avatar
Saswat Das committed
74
    "SUN397",
75
76
77
78
79
80
81
    "SVHN",
    "PhotoTour",
    "SEMEION",
    "Omniglot",
    "SBU",
    "Flickr8k",
    "Flickr30k",
Zhiqiang Wang's avatar
Zhiqiang Wang committed
82
    "Flowers102",
83
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",
    "Kinetics400",
    "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
    "CREStereo",
Ponku's avatar
Ponku committed
123
    "FallingThingsStereo",
124
    "SceneFlowStereo",
Ponku's avatar
Ponku committed
125
    "SintelStereo",
Ponku's avatar
Ponku committed
126
    "InStereo2k",
Ponku's avatar
Ponku committed
127
    "ETH3DStereo",
128
)