Source code for super_gradients.training.datasets.detection_datasets.coco_detection

from super_gradients.training.datasets.detection_datasets.detection_dataset import DetectionDataSet
from super_gradients.training.datasets.datasets_conf import COCO_DETECTION_CLASSES_LIST


[docs]class COCODetectionDataSet(DetectionDataSet): """ COCODetectionDataSet - Detection Data Set Class COCO Data Set """ def __init__(self, *args, **kwargs): kwargs['all_classes_list'] = COCO_DETECTION_CLASSES_LIST super().__init__(*args, **kwargs)