__init__.py 1.76 KB
Newer Older
1
from ._optical_flow import KittiFlow, Sintel, FlyingChairs
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
6
from .cityscapes import Cityscapes
from .coco import CocoCaptions, CocoDetection
7
from .fakedata import FakeData
8
from .flickr import Flickr8k, Flickr30k
9
10
from .folder import ImageFolder, DatasetFolder
from .hmdb51 import HMDB51
Philip Meier's avatar
Philip Meier committed
11
from .imagenet import ImageNet
12
from .inaturalist import INaturalist
13
from .kinetics import Kinetics400, Kinetics
Prabhat Roy's avatar
Prabhat Roy committed
14
from .kitti import Kitti
Muhammed Abdullah's avatar
Muhammed Abdullah committed
15
from .lfw import LFWPeople, LFWPairs
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from .lsun import LSUN, LSUNClass
from .mnist import MNIST, EMNIST, FashionMNIST, KMNIST, QMNIST
from .omniglot import Omniglot
from .phototour import PhotoTour
from .places365 import Places365
from .sbd import SBDataset
from .sbu import SBU
from .semeion import SEMEION
from .stl10 import STL10
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
31

32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
__all__ = (
    "LSUN",
    "LSUNClass",
    "ImageFolder",
    "DatasetFolder",
    "FakeData",
    "CocoCaptions",
    "CocoDetection",
    "CIFAR10",
    "CIFAR100",
    "EMNIST",
    "FashionMNIST",
    "QMNIST",
    "MNIST",
    "KMNIST",
    "STL10",
    "SVHN",
    "PhotoTour",
    "SEMEION",
    "Omniglot",
    "SBU",
    "Flickr8k",
    "Flickr30k",
    "VOCSegmentation",
    "VOCDetection",
    "Cityscapes",
    "ImageNet",
    "Caltech101",
    "Caltech256",
    "CelebA",
    "WIDERFace",
    "SBDataset",
    "VisionDataset",
    "USPS",
    "Kinetics400",
    "Kinetics",
    "HMDB51",
    "UCF101",
    "Places365",
    "Kitti",
    "INaturalist",
    "LFWPeople",
    "LFWPairs",
75
76
    "KittiFlow",
    "Sintel",
77
    "FlyingChairs",
78
)