Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
23

	
24

	
25

	
26

	
27

	
28

	
            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]