supervisely 6.73.276__py3-none-any.whl → 6.73.278__py3-none-any.whl

supervisely/__init__.py

@@ -268,6 +268,8 @@ import inspect
 from supervisely.task.progress import tqdm_sly
 import tqdm
 
+from supervisely import convert
+
 _original_tqdm = tqdm.tqdm
 
 

supervisely/annotation/annotation.py

@@ -9,7 +9,7 @@ import json
 import operator
 from collections import defaultdict
 from copy import deepcopy
-from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Tuple, Union
+from typing import TYPE_CHECKING, Callable, Dict, List, Literal, Optional, Tuple, Union
 
 import numpy as np
 from PIL import Image
@@ -3002,3 +3002,140 @@ class Annotation:
     #     new_labels = [label._to_subpixel_coordinate_system() for label in new_ann.labels]
     #     new_ann._labels = new_labels
     #     return new_ann
+
+    def to_coco(
+        self,
+        coco_image_id: int,
+        class_mapping: Dict[str, int],
+        coco_ann: Optional[Union[Dict, List]] = None,
+        last_label_id: Optional[int] = None,
+        coco_captions: Optional[Union[Dict, List]] = None,
+        last_caption_id: Optional[int] = None,
+    ) -> Tuple[List, List]:
+        """
+        Convert Supervisely annotation to COCO format annotation ("annotations" field).
+
+        :param coco_image_id: Image id in COCO format.
+        :type coco_image_id: int
+        :param class_mapping: Dictionary that maps class names to class ids.
+        :type class_mapping: Dict[str, int]
+        :param coco_ann: COCO annotation in dictionary or list format to append new annotations.
+        :type coco_ann: Union[Dict, List], optional
+        :param last_label_id: Last label id in COCO format to continue counting.
+        :type last_label_id: int, optional
+        :param coco_captions: COCO captions in dictionary or list format to append new captions.
+        :type coco_captions: Union[Dict, List], optional
+        :return: Tuple with list of COCO objects and list of COCO captions.
+        :rtype: :class:`tuple`
+
+
+        :Usage example:
+
+         .. code-block:: python
+
+            import supervisely as sly
+
+
+            coco_instances = dict(
+                info=dict(
+                    description="COCO dataset converted from Supervisely",
+                    url="None",
+                    version=str(1.0),
+                    year=2025,
+                    contributor="Supervisely",
+                    date_created="2025-01-01 00:00:00",
+                ),
+                licenses=[dict(url="None", id=0, name="None")],
+                images=[],
+                annotations=[],
+                categories=get_categories_from_meta(meta),  # [{"supercategory": "lemon", "id": 0, "name": "lemon"}, ...]
+            )
+
+            ann = sly.Annotation.from_json(ann_json, meta)
+            image_id = 11
+            label_id = 222
+            class_mapping = {obj_cls.name: idx for idx, obj_cls in enumerate(meta.obj_classes)}
+
+            curr_coco_ann, _ = ann.to_coco(image_id, class_mapping, coco_instances, label_id)
+            # or
+            # curr_coco_ann, _ = ann.to_coco(image_id, class_mapping, label_id=label_id)
+            # coco_instances["annotations"].extend(curr_coco_ann)
+
+            label_id += len(curr_coco_ann)
+            image_id += 1
+        """
+
+        from supervisely.convert.image.coco.coco_helper import sly_ann_to_coco
+
+        return sly_ann_to_coco(
+            ann=self,
+            coco_image_id=coco_image_id,
+            class_mapping=class_mapping,
+            coco_ann=coco_ann,
+            last_label_id=last_label_id,
+            coco_captions=coco_captions,
+            last_caption_id=last_caption_id,
+        )
+
+    def to_yolo(
+        self,
+        class_names: List[str],
+        task_type: Literal["detection", "segmentation", "pose"] = "detection",
+    ) -> List[str]:
+        """
+        Convert Supervisely annotation to YOLO annotation format.
+        Returns a list of strings, each string represents one object.
+
+        :param class_names: List of class names.
+        :type class_names: List[str]
+        :param task_type: Task type, one of "detection", "segmentation", "pose".
+        :type task_type: str
+        :return: List of objects in YOLO format.
+        :rtype: :class:`list`
+
+        :Usage example:
+
+         .. code-block:: python
+
+            import supervisely as sly
+
+            ann = sly.Annotation.from_json(ann_json, meta)
+            class_names = [obj_cls.name for obj_cls in meta.obj_classes]
+
+            yolo_lines = ann.to_yolo(class_names, task_type="segmentation")
+        """
+
+        from supervisely.convert.image.yolo.yolo_helper import sly_ann_to_yolo
+
+        return sly_ann_to_yolo(ann=self, class_names=class_names, task_type=task_type)
+
+    def to_pascal_voc(
+        self,
+        image_name: str,
+    ) -> Tuple[List, List]:
+        """
+        Convert Supervisely annotation to Pascal VOC format annotation ("annotations" field).
+
+        :param ann: Supervisely annotation.
+        :type ann: :class:`Annotation<supervisely.annotation.annotation.Annotation>`
+        :param image_name: Image name.
+        :type image_name: :class:`str`
+        :return: Tuple with xml tree and instance and class masks in PIL.Image format.
+        :rtype: :class:`Tuple`
+
+        :Usage example:
+
+        .. code-block:: python
+
+            import supervisely as sly
+            from supervisely.convert.image.pascal_voc.pascal_voc_helper import sly_ann_to_pascal_voc
+
+            ann = sly.Annotation.from_json(ann_json, meta)
+            xml_tree, instance_mask, class_mask = sly_ann_to_pascal_voc(ann, image_name)
+        """
+
+        from supervisely.convert.image.pascal_voc.pascal_voc_helper import (
+            sly_ann_to_pascal_voc,
+        )
+
+        return sly_ann_to_pascal_voc(self, image_name)

supervisely/convert/__init__.py

@@ -1,59 +1,16 @@
 # isort: skip_file
-from supervisely.convert.base_converter import (
-    AvailableImageConverters,
-    AvailablePointcloudConverters,
-    AvailableVideoConverters,
-    AvailableVolumeConverters,
-    BaseConverter,
-)
-from supervisely.convert.converter import ImportManager
 
-# Image
-from supervisely.convert.image.cityscapes.cityscapes_converter import (
-    CityscapesConverter,
-)
-from supervisely.convert.image.coco.coco_converter import COCOConverter
-from supervisely.convert.image.coco.coco_anntotation_converter import FastCOCOConverter
-from supervisely.convert.image.csv.csv_converter import CSVConverter
-from supervisely.convert.image.multispectral.multispectral_converter import (
-    MultiSpectralImageConverter,
-)
-from supervisely.convert.image.medical2d.medical2d_converter import Medical2DImageConverter
-from supervisely.convert.image.masks.images_with_masks_converter import ImagesWithMasksConverter
+# Project
+from supervisely.convert.image.coco.coco_helper import sly_project_to_coco as project_to_coco
+from supervisely.convert.image.yolo.yolo_helper import sly_project_to_yolo as project_to_yolo
+from supervisely.convert.image.pascal_voc.pascal_voc_helper import sly_project_to_pascal_voc as project_to_pascal_voc
 
-from supervisely.convert.image.pascal_voc.pascal_voc_converter import PascalVOCConverter
-from supervisely.convert.image.pdf.pdf_converter import PDFConverter
-from supervisely.convert.image.sly.sly_image_converter import SLYImageConverter
-from supervisely.convert.image.sly.fast_sly_image_converter import FastSlyImageConverter
-from supervisely.convert.image.yolo.yolo_converter import YOLOConverter
-from supervisely.convert.image.multi_view.multi_view import MultiViewImageConverter
-from supervisely.convert.image.label_me.label_me_converter import LabelmeConverter
-from supervisely.convert.image.label_studio.label_studio_converter import LabelStudioConverter
-from supervisely.convert.image.high_color.high_color_depth import HighColorDepthImageConverter
+# Dataset
+from supervisely.convert.image.coco.coco_helper import sly_ds_to_coco as dataset_to_coco
+from supervisely.convert.image.yolo.yolo_helper import sly_ds_to_yolo as dataset_to_yolo
+from supervisely.convert.image.pascal_voc.pascal_voc_helper import sly_ds_to_pascal_voc as dataset_to_pascal_voc
 
-
-# Pointcloud
-from supervisely.convert.pointcloud.sly.sly_pointcloud_converter import SLYPointcloudConverter
-from supervisely.convert.pointcloud.las.las_converter import LasConverter
-from supervisely.convert.pointcloud.ply.ply_converter import PlyConverter
-from supervisely.convert.pointcloud.bag.bag_converter import BagConverter
-from supervisely.convert.pointcloud.lyft.lyft_converter import LyftConverter
-from supervisely.convert.pointcloud.nuscenes_conv.nuscenes_converter import NuscenesConverter
-
-# Pointcloud Episodes
-from supervisely.convert.pointcloud_episodes.sly.sly_pointcloud_episodes_converter import (
-    SLYPointcloudEpisodesConverter,
-)
-from supervisely.convert.pointcloud_episodes.bag.bag_converter import BagEpisodesConverter
-from supervisely.convert.pointcloud_episodes.lyft.lyft_converter import LyftEpisodesConverter
-from supervisely.convert.pointcloud_episodes.nuscenes_conv.nuscenes_converter import (
-    NuscenesEpisodesConverter,
-)
-
-# Video
-from supervisely.convert.video.mot.mot_converter import MOTConverter
-from supervisely.convert.video.sly.sly_video_converter import SLYVideoConverter
-
-# Volume
-from supervisely.convert.volume.sly.sly_volume_converter import SLYVolumeConverter
-from supervisely.convert.volume.dicom.dicom_converter import DICOMConverter
+# Image Annotations
+from supervisely.convert.image.coco.coco_helper import sly_ann_to_coco as annotation_to_coco
+from supervisely.convert.image.yolo.yolo_helper import sly_ann_to_yolo as annotation_to_yolo
+from supervisely.convert.image.pascal_voc.pascal_voc_helper import sly_ann_to_pascal_voc as annotation_to_pascal_voc

supervisely/convert/base_converter.py

@@ -19,6 +19,8 @@ from supervisely.io.fs import (
     silent_remove,
     unpack_archive,
 )
+from supervisely.annotation.obj_class import ObjClass
+from supervisely.geometry.graph import GraphNodes
 from supervisely.project.project_meta import ProjectMeta
 from supervisely.project.project_settings import LabelingInterface
 from supervisely.sly_logger import logger
@@ -56,6 +58,7 @@ class AvailablePointcloudConverters:
     BAG = "rosbag"
     LYFT = "lyft"
     NUSCENES = "nuscenes"
+    KITTI3D = "kitti3d"
 
 
 class AvailablePointcloudEpisodesConverters:
@@ -343,8 +346,17 @@ class BaseConverter:
             i = 1
             new_name = new_cls.name
             matched = False
+            def _is_matched(old_cls: ObjClass, new_cls: ObjClass) -> bool:
+                if old_cls.geometry_type == new_cls.geometry_type:
+                    if old_cls.geometry_type == GraphNodes:
+                        old_nodes = old_cls.geometry_config["nodes"]
+                        new_nodes = new_cls.geometry_config["nodes"]
+                        return old_nodes.keys() == new_nodes.keys()
+                    return True
+                return False
+
             while meta1.obj_classes.get(new_name) is not None:
-                if meta1.obj_classes.get(new_name).geometry_type == new_cls.geometry_type:
+                if _is_matched(meta1.get_obj_class(new_name), new_cls):
                     matched = True
                     break
                 new_name = f"{new_cls.name}_{i}"

supervisely/convert/image/__init__.py

@@ -0,0 +1,22 @@
+# Image
+from supervisely.convert.image.cityscapes.cityscapes_converter import (
+    CityscapesConverter,
+)
+from supervisely.convert.image.coco.coco_converter import COCOConverter
+from supervisely.convert.image.coco.coco_anntotation_converter import FastCOCOConverter
+from supervisely.convert.image.csv.csv_converter import CSVConverter
+from supervisely.convert.image.multispectral.multispectral_converter import (
+    MultiSpectralImageConverter,
+)
+from supervisely.convert.image.medical2d.medical2d_converter import Medical2DImageConverter
+from supervisely.convert.image.masks.images_with_masks_converter import ImagesWithMasksConverter
+
+from supervisely.convert.image.pascal_voc.pascal_voc_converter import PascalVOCConverter
+from supervisely.convert.image.pdf.pdf_converter import PDFConverter
+from supervisely.convert.image.sly.sly_image_converter import SLYImageConverter
+from supervisely.convert.image.sly.fast_sly_image_converter import FastSlyImageConverter
+from supervisely.convert.image.yolo.yolo_converter import YOLOConverter
+from supervisely.convert.image.multi_view.multi_view import MultiViewImageConverter
+from supervisely.convert.image.label_me.label_me_converter import LabelmeConverter
+from supervisely.convert.image.label_studio.label_studio_converter import LabelStudioConverter
+from supervisely.convert.image.high_color.high_color_depth import HighColorDepthImageConverter