You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
|
|
# Copyright (C) 2019 Intel Corporation
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from datumaro.components.converters.datumaro import DatumaroConverter
|
|
|
|
from datumaro.components.converters.ms_coco import (
|
|
CocoConverter,
|
|
CocoImageInfoConverter,
|
|
CocoCaptionsConverter,
|
|
CocoInstancesConverter,
|
|
CocoPersonKeypointsConverter,
|
|
CocoLabelsConverter,
|
|
)
|
|
|
|
from datumaro.components.converters.voc import (
|
|
VocConverter,
|
|
VocClassificationConverter,
|
|
VocDetectionConverter,
|
|
VocLayoutConverter,
|
|
VocActionConverter,
|
|
VocSegmentationConverter,
|
|
)
|
|
|
|
from datumaro.components.converters.yolo import YoloConverter
|
|
from datumaro.components.converters.tfrecord import DetectionApiConverter
|
|
|
|
|
|
items = [
|
|
('datumaro', DatumaroConverter),
|
|
|
|
('coco', CocoConverter),
|
|
('coco_images', CocoImageInfoConverter),
|
|
('coco_captions', CocoCaptionsConverter),
|
|
('coco_instances', CocoInstancesConverter),
|
|
('coco_person_kp', CocoPersonKeypointsConverter),
|
|
('coco_labels', CocoLabelsConverter),
|
|
|
|
('voc', VocConverter),
|
|
('voc_cls', VocClassificationConverter),
|
|
('voc_det', VocDetectionConverter),
|
|
('voc_segm', VocSegmentationConverter),
|
|
('voc_action', VocActionConverter),
|
|
('voc_layout', VocLayoutConverter),
|
|
|
|
('yolo', YoloConverter),
|
|
|
|
('tf_detection_api', DetectionApiConverter),
|
|
]
|