This package is intended to be a multi-type importer/exporter/converter between different annotation formats.
This package is currently in development and has not that many features implemented. The API is not stable and is subject to change.
The package consists of the Intermediary Representation (IR) annotation format in Python Objects, and importers/exporters for different annotation formats.
pip install dagshub-annotation-converter- YOLO BBox, Segmentation and Poses:
load_yolo_from_fs - COCO BBox Object Detection and Stuff Segmentation:
load_coco_from_file - Label Studio:
parse_ls_task(Only reading JSON tasks from the filesystem is implemented, importing from a project is left up to user):
from dagshub_annotation_converter.formats.label_studio.task import LabelStudioTask
task_obj = LabelStudioTask.from_json("path/to/label_studio_task.json")
annotations = task_obj.to_ir_annotations()- CVAT for image (BBox, Segmentation, Poses) and video (BBox):
load_cvat_from_zip - MOT (Video BBox) -
load_mot_from_dir
- YOLO BBox, Segmentation and Poses:
export_to_fs - COCO BBox Object Detection and Stuff Segmentation:
export_to_coco_file - Label Studio:
task.model_dump_json()(Again, only to JSON files, uploading the task to the project is left to the user) - MOT (Video BBox) -
export_mot_to_dir - CVAT for video (BBox):
export_cvat_video_to_zip
The package contains some experimental features that can be toggled on by setting a corresponding environment variable:
DAGSHUB_ANNOTATION_EXPERIMENTAL_CVAT_POSE_GROUPING_BY_GROUP_ID_ENABLED- When importing CVAT annotations, you can put a single bbox and either a points or skeletons annotation in a group, and then they will be grouped together into a pose annotation. When exporting that to YOLO, the bbox and the points will be exported as a single annotation.
You can read more about the features that exist, their limitations and potential side effects in the features.py file