supervisely 6.73.300__tar.gz → 6.73.302__tar.gz

{supervisely-6.73.300/supervisely.egg-info → supervisely-6.73.302}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: supervisely
-Version: 6.73.300
+Version: 6.73.302
 Summary: Supervisely Python SDK.
 Home-page: https://github.com/supervisely/supervisely
 Author: Supervisely

{supervisely-6.73.300 → supervisely-6.73.302}/supervisely/convert/__init__.py

@@ -3,14 +3,30 @@
 # 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_helper import (
+    sly_project_to_pascal_voc as project_to_pascal_voc,
+)
 
 # 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
+from supervisely.convert.image.pascal_voc.pascal_voc_helper import (
+    sly_ds_to_pascal_voc as dataset_to_pascal_voc,
+)
 
 # 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
+from supervisely.convert.image.pascal_voc.pascal_voc_helper import (
+    sly_ann_to_pascal_voc as annotation_to_pascal_voc,
+)
+
+
+# Supervisely Project/Dataset/Annotation to COCO
+from supervisely.convert.image.coco.coco_helper import to_coco
+
+# Supervisely Project/Dataset/Annotation to YOLO
+from supervisely.convert.image.yolo.yolo_helper import to_yolo
+
+# Supervisely Project/Dataset/Annotation to Pascal VOC
+from supervisely.convert.image.pascal_voc.pascal_voc_helper import to_pascal_voc

{supervisely-6.73.300 → supervisely-6.73.302}/supervisely/convert/image/coco/coco_helper.py

@@ -4,6 +4,7 @@ import sys
 import uuid
 from copy import deepcopy
 from datetime import datetime
+from itertools import groupby
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Literal, Optional, Tuple, Union
 
@@ -351,7 +352,8 @@ def _get_graph_info(idx, obj_class):
     return data
 
 
-def get_categories_from_meta(meta: ProjectMeta):
+def get_categories_from_meta(meta: ProjectMeta) -> List[Dict[str, Any]]:
+    """Get categories from Supervisely project meta."""
     cat = lambda idx, c: {"supercategory": c.name, "id": idx, "name": c.name}
     return [
         cat(idx, c) if c.geometry_type != GraphNodes else _get_graph_info(idx, c)
@@ -359,6 +361,34 @@ def get_categories_from_meta(meta: ProjectMeta):
     ]
 
 
+def extend_mask_up_to_image(
+    binary_mask: np.ndarray, image_shape: Tuple[int, int], origin: PointLocation
+) -> np.ndarray:
+    """Extend binary mask up to image shape."""
+    y, x = origin.col, origin.row
+    new_mask = np.zeros(image_shape, dtype=binary_mask.dtype)
+    try:
+        new_mask[x : x + binary_mask.shape[0], y : y + binary_mask.shape[1]] = binary_mask
+    except ValueError as e:
+        raise ValueError(
+            f"Binary mask size {binary_mask.shape} with origin {origin} "
+            f"exceeds image boundaries {image_shape}"
+        ) from e
+    return new_mask
+
+
+def coco_segmentation_rle(segmentation: np.ndarray) -> Dict[str, Any]:
+    """Convert COCO segmentation to RLE format."""
+    binary_mask = np.asfortranarray(segmentation)
+    rle = {"counts": [], "size": list(binary_mask.shape)}
+    counts = rle.get("counts")
+    for i, (value, elements) in enumerate(groupby(binary_mask.ravel(order="F"))):
+        if i == 0 and value == 1:
+            counts.append(0)
+        counts.append(len(list(elements)))
+    return rle
+
+
 def sly_ann_to_coco(
     ann: Annotation,
     coco_image_id: int,
@@ -491,17 +521,14 @@ def sly_ann_to_coco(
 
     res_inst = []  # result list of COCO objects
 
+    h, w = ann.img_size
     for binding_key, labels in ann.get_bindings().items():
         if binding_key is None:
             polygons = [l for l in labels if l.obj_class.geometry_type == Polygon]
             masks = [l for l in labels if l.obj_class.geometry_type == Bitmap]
             bboxes = [l for l in labels if l.obj_class.geometry_type == Rectangle]
             graphs = [l for l in labels if l.obj_class.geometry_type == GraphNodes]
-            if len(masks) > 0:
-                for l in masks:
-                    polygon_cls = l.obj_class.clone(geometry_type=Polygon)
-                    polygons.extend(l.convert(polygon_cls))
-            for label in polygons + bboxes:
+            for label in polygons + bboxes + masks:
                 cat_id = class_mapping[label.obj_class.name]
                 coco_obj = coco_obj_template(label_id, coco_image_id, cat_id)
                 coco_obj["bbox"] = _get_common_bbox([label])
@@ -510,6 +537,11 @@ def sly_ann_to_coco(
                     poly = label.geometry.to_json()["points"]["exterior"]
                     poly = np.array(poly).flatten().astype(float).tolist()
                     coco_obj["segmentation"] = [poly]
+                elif label.obj_class.geometry_type == Bitmap:
+                    segmentation = extend_mask_up_to_image(
+                        label.geometry.data, (h, w), label.geometry.origin
+                    )
+                    coco_obj["segmentation"] = coco_segmentation_rle(segmentation)
 
                 label_id = _update_inst_results(label_id, coco_ann, coco_obj, res_inst)
 
@@ -525,38 +557,60 @@ def sly_ann_to_coco(
         masks = [l for l in labels if l.obj_class.geometry_type == Bitmap]
         graphs = [l for l in labels if l.obj_class.geometry_type == GraphNodes]
 
-        if len(masks) > 0:  # convert Bitmap to Polygon
-            for l in masks:
-                polygon_cls = l.obj_class.clone(geometry_type=Polygon)
-                polygons.extend(l.convert(polygon_cls))
+        need_to_process_separately = len(masks) > 0 and len(polygons) > 0
+        bbox_matched_w_mask = False
+        bbox_matched_w_poly = False
 
-        matched_bbox = False
-        if len(polygons) > 0:  # process polygons
+        if len(graphs) > 0:
+            if len(masks) > 0 or len(polygons) > 0:
+                logger.warning(
+                    "Keypoints and Polygons/Bitmaps in one binding key are not supported. "
+                    "Objects will be converted separately."
+                )
+            if len(graphs) > 1:
+                logger.warning(
+                    "Multiple Keypoints in one binding key are not supported. "
+                    "Only the first graph will be converted."
+                )
+            cat_id = class_mapping[graphs[0].obj_class.name]
+            coco_obj = _create_keypoints_obj(graphs[0], cat_id, label_id, coco_image_id)
+            label_id = _update_inst_results(label_id, coco_ann, coco_obj, res_inst)
+
+        # convert Bitmap to Polygon
+        if len(masks) > 0:
+            for label in masks:
+                cat_id = class_mapping[label.obj_class.name]
+                coco_obj = coco_obj_template(label_id, coco_image_id, cat_id)
+                segmentation = extend_mask_up_to_image(
+                    label.geometry.data, (h, w), label.geometry.origin
+                )
+                coco_obj["segmentation"] = coco_segmentation_rle(segmentation)
+                coco_obj["area"] = label.geometry.area
+                if len(bboxes) > 0 and not need_to_process_separately:
+                    found = _get_common_bbox(bboxes, sly_bbox=True, approx=True)
+                    new = _get_common_bbox([label], sly_bbox=True)
+                    bbox_matched_w_mask = found.contains(new)
+                coco_obj["bbox"] = _get_common_bbox(bboxes if bbox_matched_w_mask else [label])
+                label_id = _update_inst_results(label_id, coco_ann, coco_obj, res_inst)
+
+        # process polygons
+        if len(polygons) > 0:
             cat_id = class_mapping[polygons[0].obj_class.name]
             coco_obj = coco_obj_template(label_id, coco_image_id, cat_id)
-            if len(bboxes) > 0:
+            if len(bboxes) > 0 and not need_to_process_separately:
                 found = _get_common_bbox(bboxes, sly_bbox=True, approx=True)
                 new = _get_common_bbox(polygons, sly_bbox=True)
-                matched_bbox = found.contains(new)
+                bbox_matched_w_poly = found.contains(new)
 
             polys = [l.geometry.to_json()["points"]["exterior"] for l in polygons]
             polys = [np.array(p).flatten().astype(float).tolist() for p in polys]
             coco_obj["segmentation"] = polys
             coco_obj["area"] = sum([l.geometry.area for l in polygons])
-            coco_obj["bbox"] = _get_common_bbox(bboxes if matched_bbox else polygons)
-            label_id = _update_inst_results(label_id, coco_ann, coco_obj, res_inst)
-
-        if len(graphs) > 0:
-            if len(graphs) > 1:
-                logger.warning(
-                    "Multiple Keypoints in one binding key are not supported. "
-                    "Only the first graph will be converted."
-                )
-            cat_id = class_mapping[graphs[0].obj_class.name]
-            coco_obj = _create_keypoints_obj(graphs[0], cat_id, label_id, coco_image_id)
+            coco_obj["bbox"] = _get_common_bbox(bboxes if bbox_matched_w_poly else polygons)
             label_id = _update_inst_results(label_id, coco_ann, coco_obj, res_inst)
 
-        if len(bboxes) > 0 and not matched_bbox:  # process bboxes separately
+        # process bboxes separately if they are not matched with masks/polygons
+        if len(bboxes) > 0 and not bbox_matched_w_poly and not bbox_matched_w_mask:
             for label in bboxes:
                 cat_id = class_mapping[label.obj_class.name]
                 coco_obj = coco_obj_template(label_id, coco_image_id, cat_id)
@@ -713,7 +767,9 @@ def sly_ds_to_coco(
 
         coco_ann["images"].append(image_coco(image_info, image_idx))
         if with_captions is True:
-            coco_captions["images"].append(image_coco(image_info, image_idx))  # pylint: disable=unsubscriptable-object
+            # pylint: disable=unsubscriptable-object
+            coco_captions["images"].append(image_coco(image_info, image_idx))
+            # pylint: enable=unsubscriptable-object
 
         ann = Annotation.load_json_file(ann_path, meta)
         if ann.img_size is None or ann.img_size == (0, 0) or ann.img_size == (None, None):
@@ -815,3 +871,97 @@ def sly_project_to_coco(
         )
         logger.info(f"Dataset '{ds.short_name}' has been converted to COCO format.")
     logger.info(f"Project '{project.name}' has been converted to COCO format.")
+
+
+def to_coco(
+    input_data: Union[Project, Dataset, str],
+    dest_dir: Optional[str] = None,
+    meta: Optional[ProjectMeta] = None,
+    copy_images: bool = True,
+    with_captions: bool = False,
+    log_progress: bool = True,
+    progress_cb: Optional[Callable] = None,
+) -> Union[None, str]:
+    """
+    Universal function to convert Supervisely project or dataset to COCO format.
+    Note:
+        - For better compatibility, please pass named arguments explicitly. Otherwise, the function may not work as expected.
+            You can use the dedicated functions for each data type:
+
+                - :func:`sly.convert.sly_project_to_coco()`
+                - :func:`sly.convert.sly_ds_to_coco()`
+
+        - If the input_data is a Project, the dest_dir parameters are required.
+        - If the input_data is a Dataset, the meta and dest_dir parameters are required.
+
+    :param input_data: Supervisely project, dataset, or path to the project or dataset.
+    :type input_data: :class:`Project<supervisely.project.project.Project>`, :class:`Dataset<supervisely.project.dataset.Dataset>` or :class:`str`
+
+    # Project or Dataset conversion arguments:
+    :param dest_dir: Destination directory to save project or dataset in COCO format.
+    :type dest_dir: :class:`str`, optional
+    :param meta: Project meta information (required for dataset conversion).
+    :type meta: :class:`ProjectMeta<supervisely.project.project_meta.ProjectMeta>`, optional
+    :param copy_images: Copy images to destination directory
+    :type copy_images: :class:`bool`, optional
+    :param with_captions: If True, returns COCO captions
+    :type with_captions: :class:`bool`, optional
+    :param log_progress: Show uploading progress bar
+    :type log_progress: :class:`bool`
+    :param progress_cb: Function for tracking conversion progress (for all items in the project or dataset).
+    :type progress_cb: callable, optional
+
+    :return: None
+    :rtype: NoneType
+
+    :Usage example:
+
+    .. code-block:: python
+
+        import supervisely as sly
+
+        # Local folder with Project in Supervisely format
+        project_directory = "./source/project"
+        project_fs = sly.Project(project_directory, sly.OpenMode.READ)
+
+        # Convert Project to COCO format
+        sly.convert.to_coco(project_directory, dest_dir="./coco")
+        # or
+        sly.convert.to_coco(project_fs, dest_dir="./coco")
+
+        # Convert Dataset to COCO format
+        # dataset: sly.Dataset
+        sly.convert.to_coco(dataset, dest_dir="./coco", meta=project_fs.meta)
+    """
+    if isinstance(input_data, str):
+        try:
+            input_data = Project(input_data, mode=OpenMode.READ)
+        except Exception:
+            try:
+                input_data = Dataset(input_data, mode=OpenMode.READ)
+            except Exception:
+                raise ValueError("Please check the path or the input data.")
+
+    if isinstance(input_data, Project):
+        return sly_project_to_coco(
+            project=input_data,
+            dest_dir=dest_dir,
+            copy_images=copy_images,
+            with_captions=with_captions,
+            log_progress=log_progress,
+            progress_cb=progress_cb,
+        )
+    if isinstance(input_data, Dataset):
+        if meta is None:
+            raise ValueError("Project meta information is required for dataset conversion.")
+        return sly_ds_to_coco(
+            dataset=input_data,
+            meta=meta,
+            return_type="path",
+            dest_dir=dest_dir,
+            copy_images=copy_images,
+            with_captions=with_captions,
+            log_progress=log_progress,
+            progress_cb=progress_cb,
+        )
+    raise ValueError("Unsupported input type. Only Project or Dataset are supported.")

{supervisely-6.73.300 → supervisely-6.73.302}/supervisely/convert/image/pascal_voc/pascal_voc_helper.py

@@ -7,17 +7,11 @@ import numpy as np
 from PIL import Image
 from tqdm import tqdm
 
-from supervisely import (
-    Annotation,
-    Dataset,
-    Label,
-    ObjClass,
-    ObjClassCollection,
-    Project,
-    ProjectMeta,
-    generate_free_name,
-    logger,
-)
+from supervisely._utils import generate_free_name
+from supervisely.annotation.annotation import Annotation
+from supervisely.annotation.label import Label
+from supervisely.annotation.obj_class import ObjClass
+from supervisely.annotation.obj_class_collection import ObjClassCollection
 from supervisely.convert.image.image_helper import validate_image_bounds
 from supervisely.geometry.bitmap import Bitmap
 from supervisely.geometry.polygon import Polygon
@@ -26,6 +20,9 @@ from supervisely.imaging.color import generate_rgb
 from supervisely.imaging.image import read
 from supervisely.io.fs import file_exists, get_file_ext, get_file_name
 from supervisely.io.json import load_json_file
+from supervisely.project.project import Dataset, OpenMode, Project
+from supervisely.project.project_meta import ProjectMeta
+from supervisely.sly_logger import logger
 from supervisely.task.progress import tqdm_sly
 
 MASKS_EXTENSION = ".png"
@@ -374,7 +371,7 @@ def sly_ds_to_pascal_voc(
     train_val_split_coef: float = 0.8,
     log_progress: bool = False,
     progress_cb: Optional[Union[tqdm, Callable]] = None,
-) -> Tuple[Dict, Optional[Dict]]:
+) -> None:
     """
     Convert Supervisely dataset to Pascal VOC format.
 
@@ -612,6 +609,9 @@ def sly_project_to_pascal_voc(
         # Convert Project to Pascal VOC format
         sly.Project(project_directory).to_pascal_voc(log_progress=True)
     """
+    if isinstance(project, str):
+        project = Project(project, mode=OpenMode.READ)
+
     if dest_dir is None:
         dest_dir = project.directory
 
@@ -636,3 +636,87 @@ def sly_project_to_pascal_voc(
         )
         logger.info(f"Dataset '{dataset.short_name}' has been converted to Pascal VOC format.")
     logger.info(f"Project '{project.name}' has been converted to Pascal VOC format.")
+
+
+def to_pascal_voc(
+    input_data: Union[Project, Dataset, str],
+    dest_dir: Optional[str] = None,
+    meta: Optional[ProjectMeta] = None,
+    train_val_split_coef: float = 0.8,
+    log_progress: bool = True,
+    progress_cb: Optional[Union[tqdm, Callable]] = None,
+) -> None:
+    """
+    Universal function to convert Supervisely project or dataset to Pascal VOC format.
+    Note:
+        - For better compatibility, please pass named arguments explicitly. Otherwise, the function may not work as expected.
+            You can use the dedicated functions for each data type:
+
+                - :func:`sly.convert.sly_project_to_pascal_voc()`
+                - :func:`sly.convert.sly_ds_to_pascal_voc()`
+
+        - If the input_data is a Project, the dest_dir parameters are required.
+        - If the input_data is a Dataset, the meta and dest_dir parameters are required.
+
+    :param input_data: Input data to convert (Project, Dataset, or path to the project/dataset directory).
+    :type input_data: :class:`Project<supervisely.project.project.Project>`, :class:`Dataset<supervisely.dataset.dataset.Dataset>`, or :class:`str`
+    :param dest_dir: Destination directory.
+    :type dest_dir: :class:`str`, optional
+    :param meta: Project meta information (required for Dataset conversion).
+    :type meta: :class:`ProjectMeta<supervisely.project.project_meta.ProjectMeta>`, optional
+    :param train_val_split_coef: Coefficient for splitting images into train and validation sets.
+    :type train_val_split_coef: :class:`float`, optional
+    :param log_progress: Show uploading progress bar.
+    :type log_progress: :class:`bool`
+    :param progress_cb: Function for tracking conversion progress (for all items in the project).
+    :type progress_cb: callable, optional
+    :return: None
+    :rtype: NoneType
+
+    :Usage example:
+
+    .. code-block:: python
+
+        import supervisely as sly
+
+        # Local folder with Project
+        project_directory = "/home/admin/work/supervisely/source/project"
+        project_fs = sly.Project(project_directory, sly.OpenMode.READ)
+
+        # Convert Project to Pascal VOC format
+        sly.convert.to_pascal_voc(project_directory, dest_dir="./pascal_voc")
+        # or
+        sly.convert.to_pascal_voc(project_fs, dest_dir="./pascal_voc")
+
+        # Convert Dataset to Pascal VOC format
+        dataset: sly.Dataset = project_fs.datasets.get("dataset_name")
+        sly.convert.to_pascal_voc(dataset, dest_dir="./pascal_voc")
+    """
+    if isinstance(input_data, str):
+        try:
+            input_data = Project(input_data, mode=OpenMode.READ)
+        except Exception as e:
+            try:
+                input_data = Dataset(input_data, mode=OpenMode.READ)
+            except Exception as e:
+                raise ValueError("Please check the path or the input data.")
+
+    if isinstance(input_data, (Project, str)):
+        return sly_project_to_pascal_voc(
+            project=input_data,
+            dest_dir=dest_dir,
+            train_val_split_coef=train_val_split_coef,
+            log_progress=log_progress,
+            progress_cb=progress_cb,
+        )
+    elif isinstance(input_data, Dataset):
+        return sly_ds_to_pascal_voc(
+            dataset=input_data,
+            meta=meta,
+            dest_dir=dest_dir,
+            train_val_split_coef=train_val_split_coef,
+            log_progress=log_progress,
+            progress_cb=progress_cb,
+        )
+    else:
+        raise ValueError(f"Unsupported input data type: {type(input_data)}")

{supervisely-6.73.300 → supervisely-6.73.302}/supervisely/convert/image/yolo/yolo_helper.py

@@ -474,7 +474,7 @@ def label_to_yolo_lines(
                 max_kpts_count=max_kpts_count,
             )
         else:
-            raise ValueError(f"Unsupported geometry type: {type(label.obj_class.geometry_type)}")
+            raise ValueError(f"Unsupported task type: {task_type}")
 
         if yolo_line is not None:
             lines.append(yolo_line)
@@ -657,3 +657,86 @@ def sly_project_to_yolo(
         )
         logger.info(f"Dataset '{dataset.short_name}' has been converted to YOLO format.")
     logger.info(f"Project '{project.name}' has been converted to YOLO format.")
+
+
+def to_yolo(
+    input_data: Union[Project, Dataset, str],
+    dest_dir: Optional[str] = None,
+    task_type: Literal["detection", "segmentation", "pose"] = "detection",
+    meta: Optional[ProjectMeta] = None,
+    log_progress: bool = True,
+    progress_cb: Optional[Callable] = None,
+) -> Union[None, str]:
+    """
+    Universal function to convert Supervisely project or dataset  to YOLO format.
+    Note:
+        - For better compatibility, please pass named arguments explicitly. Otherwise, the function may not work as expected.
+            You can use the dedicated functions for each data type:
+
+                - :func:`sly.convert.sly_project_to_yolo()`
+                - :func:`sly.convert.sly_ds_to_yolo()`
+
+        - If the input_data is a Project, the dest_dir parameters are required.
+        - If the input_data is a Dataset, the meta and dest_dir parameters are required.
+
+    :param input_data: Supervisely project or dataset, or path to the directory with the project/dataset.
+    :type input_data: :class:`supervisely.project.project.Project`, :class:`supervisely.project.dataset.Dataset`, or :class:`str`
+    :param dest_dir: Destination directory.
+    :type dest_dir: :class:`str`, optional
+    :param task_type: Task type.
+    :type task_type: :class:`str`, optional
+    :param meta: Project meta (required for Dataset conversion).
+    :type meta: :class:`supervisely.project.project_meta.ProjectMeta`, optional
+    :param log_progress: Show uploading progress bar.
+    :type log_progress: :class:`bool`
+    :param progress_cb: Function for tracking conversion progress (for all items in the project).
+    :type progress_cb: callable, optional
+    :return: None, list of YOLO lines, or path to the destination directory.
+    :rtype: NoneType, list, str
+
+    :Usage example:
+
+    .. code-block:: python
+
+        import supervisely as sly
+
+        # Local folder with Project
+        project_directory = "/home/admin/work/supervisely/source/project"
+        project_fs = sly.Project(project_directory, sly.OpenMode.READ)
+
+        # Convert Project to YOLO format
+        sly.convert.to_yolo(project_directory, dest_dir="./yolo")
+        # or
+        sly.convert.to_yolo(project_fs, dest_dir="./yolo")
+
+        # Convert Dataset to YOLO format
+        dataset: sly.Dataset = project_fs.datasets.get("dataset_name")
+        sly.convert.to_yolo(dataset, dest_dir="./yolo", meta=project_fs.meta)
+    """
+    if isinstance(input_data, str):
+        try:
+            input_data = Project(input_data, mode=OpenMode.READ)
+        except Exception:
+            try:
+                input_data = Dataset(input_data, mode=OpenMode.READ)
+            except Exception:
+                raise ValueError("Please check the path or the input data.")
+    if isinstance(input_data, Project):
+        return sly_project_to_yolo(
+            project=input_data,
+            dest_dir=dest_dir,
+            task_type=task_type,
+            log_progress=log_progress,
+            progress_cb=progress_cb,
+        )
+    elif isinstance(input_data, Dataset):
+        return sly_ds_to_yolo(
+            dataset=input_data,
+            meta=meta,
+            dest_dir=dest_dir,
+            task_type=task_type,
+            log_progress=log_progress,
+            progress_cb=progress_cb,
+        )
+    else:
+        raise ValueError("Unsupported input type. Only Project or Dataset are supported.")

{supervisely-6.73.300 → supervisely-6.73.302}/supervisely/nn/benchmark/base_visualizer.py

@@ -84,7 +84,8 @@ class BaseVisualizer:
 
         with self.pbar(message="Fetching project and dataset infos", total=len(eval_results)) as p:
             for eval_result in self.eval_results:
-                self._get_eval_project_infos(eval_result)
+                if eval_result.gt_project_id is not None:
+                    self._get_eval_project_infos(eval_result)
                 p.update(1)
 
     def _get_eval_project_infos(self, eval_result):

{supervisely-6.73.300 → supervisely-6.73.302/supervisely.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: supervisely
-Version: 6.73.300
+Version: 6.73.302
 Summary: Supervisely Python SDK.
 Home-page: https://github.com/supervisely/supervisely
 Author: Supervisely