super-gradients/src/super_gradients/training/datasets/__init__.py

from __future__ import absolute_import

from super_gradients.training.datasets.data_augmentation import DataAugmentation
from super_gradients.training.datasets.sg_dataset import ListDataset, DirectoryDataSet
from super_gradients.training.datasets.all_datasets import CLASSIFICATION_DATASETS, OBJECT_DETECTION_DATASETS, \
    SEMANTIC_SEGMENTATION_DATASETS
from super_gradients.training.datasets.detection_datasets.detection_dataset import DetectionDataSet
from super_gradients.training.datasets.detection_datasets.coco_detection import COCODetectionDataSet
from super_gradients.training.datasets.segmentation_datasets.segmentation_dataset import SegmentationDataSet
from super_gradients.training.datasets.segmentation_datasets.pascal_voc_segmentation import PascalVOC2012SegmentationDataSet
from super_gradients.training.datasets.segmentation_datasets.pascal_aug_segmentation import PascalAUG2012SegmentationDataSet
from super_gradients.training.datasets.segmentation_datasets.coco_segmentation import CoCoSegmentationDataSet
from super_gradients.training.datasets.dataset_interfaces.dataset_interface import TestDatasetInterface, DatasetInterface, \
    Cifar10DatasetInterface, CoCoSegmentationDatasetInterface, CoCoDetectionDatasetInterface, \
    CoCo2014DetectionDatasetInterface, \
    PascalVOC2012SegmentationDataSetInterface, PascalAUG2012SegmentationDataSetInterface, \
    TestYoloDetectionDatasetInterface, SegmentationTestDatasetInterface, DetectionTestDatasetInterface, \
    ClassificationTestDatasetInterface, ImageNetDatasetInterface

__all__ = ['DataAugmentation', 'ListDataset', 'DirectoryDataSet', 'CLASSIFICATION_DATASETS', 'OBJECT_DETECTION_DATASETS',
           'SEMANTIC_SEGMENTATION_DATASETS', 'DetectionDataSet', 'COCODetectionDataSet', 'SegmentationDataSet',
           'PascalVOC2012SegmentationDataSet',
           'PascalAUG2012SegmentationDataSet', 'CoCoSegmentationDataSet', 'TestDatasetInterface', 'DatasetInterface',
           'Cifar10DatasetInterface', 'CoCoSegmentationDatasetInterface', 'CoCoDetectionDatasetInterface',
           'CoCo2014DetectionDatasetInterface',
           'PascalVOC2012SegmentationDataSetInterface', 'PascalAUG2012SegmentationDataSetInterface',
           'TestYoloDetectionDatasetInterface', 'DetectionTestDatasetInterface', 'ClassificationTestDatasetInterface',
           'SegmentationTestDatasetInterface',
           'ImageNetDatasetInterface']