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

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