supervisely 6.73.301__py3-none-any.whl → 6.73.303__py3-none-any.whl

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/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/nn/artifacts/artifacts.py

@@ -3,11 +3,12 @@ import string
 from abc import abstractmethod
 from collections import defaultdict
 from concurrent.futures import ThreadPoolExecutor, as_completed
+from dataclasses import fields
 from datetime import datetime
 from json import JSONDecodeError
 from os.path import dirname, join
 from time import time
-from typing import Any, Dict, List, Literal, NamedTuple
+from typing import Any, Dict, List, Literal, NamedTuple, Union
 
 import requests
 
@@ -55,6 +56,9 @@ class BaseTrainArtifacts:
         self._metadata_file_name: str = "train_info.json"
 
         self._app_name: str = None
+        self._slug = None
+        self._serve_app_name = None
+        self._serve_slug = None
         self._framework_name: str = None
         self._framework_folder: str = None
         self._weights_folder: str = None
@@ -63,6 +67,7 @@ class BaseTrainArtifacts:
         self._config_file: str = None
         self._pattern: str = None
         self._available_task_types: List[str] = []
+        self._require_runtime = False
 
     @property
     def team_id(self) -> int:
@@ -94,6 +99,36 @@ class BaseTrainArtifacts:
         """
         return self._app_name
 
+    @property
+    def slug(self):
+        """
+        Train app slug.
+
+        :return: Train app slug.
+        :rtype: str
+        """
+        return self._slug
+
+    @property
+    def serve_app_name(self):
+        """
+        Serve application name.
+
+        :return: The serve application name.
+        :rtype: str
+        """
+        return self._serve_app_name
+
+    @property
+    def serve_slug(self):
+        """
+        Serve app slug.
+
+        :return: Serve app slug.
+        :rtype: str
+        """
+        return self._serve_slug
+
     @property
     def framework_name(self):
         """
@@ -164,6 +199,16 @@ class BaseTrainArtifacts:
         """
         return self._pattern
 
+    @property
+    def require_runtime(self):
+        """
+        Whether providing runtime is required for the framework.
+
+        :return: True if runtime is required, False otherwise.
+        :rtype: bool
+        """
+        return self._require_runtime
+
     def is_valid_artifacts_path(self, path):
         """
         Check if the provided path is valid and follows specified session path pattern.
@@ -531,68 +576,68 @@ class BaseTrainArtifacts:
         logger.debug(f"Listing time: '{format(end_time - start_time, '.6f')}' sec")
         return train_infos
 
-    def get_list_experiment_info(self, sort: Literal["desc", "asc"] = "desc") -> List[TrainInfo]:
-        def build_experiment_info_from_train_info(
-            api: Api, train_info: TrainInfo
-        ) -> ExperimentInfo:
-
-            try:
-                checkpoints = []
-                for chk in train_info.checkpoints:
-                    if self.weights_folder:
-                        checkpoints.append(join(self.weights_folder, chk.name))
-                    else:
-                        checkpoints.append(chk.name)
-
-                best_checkpoint = next(
-                    (chk.name for chk in train_info.checkpoints if "best" in chk.name), None
+    def convert_train_to_experiment_info(
+        self, train_info: TrainInfo
+    ) -> Union[ExperimentInfo, None]:
+        try:
+            checkpoints = []
+            for chk in train_info.checkpoints:
+                if self.weights_folder:
+                    checkpoints.append(join(self.weights_folder, chk.name))
+                else:
+                    checkpoints.append(chk.name)
+
+            best_checkpoint = next(
+                (chk.name for chk in train_info.checkpoints if "best" in chk.name), None
+            )
+            if not best_checkpoint and checkpoints:
+                best_checkpoint = get_file_name_with_ext(checkpoints[-1])
+
+            task_info = self._api.task.get_info_by_id(train_info.task_id)
+            workspace_id = task_info["workspaceId"]
+
+            project = self._api.project.get_info_by_name(workspace_id, train_info.project_name)
+            project_id = project.id if project else None
+
+            model_files = {}
+            if train_info.config_path:
+                model_files["config"] = self.get_config_path(train_info.artifacts_folder).replace(
+                    train_info.artifacts_folder, ""
                 )
-                if not best_checkpoint and checkpoints:
-                    best_checkpoint = get_file_name_with_ext(checkpoints[-1])
-
-                task_info = api.task.get_info_by_id(train_info.task_id)
-                workspace_id = task_info["workspaceId"]
-
-                project = api.project.get_info_by_name(workspace_id, train_info.project_name)
-                project_id = project.id if project else None
-
-                model_files = {}
-                if train_info.config_path:
-                    model_files["config"] = self.get_config_path(
-                        train_info.artifacts_folder
-                    ).replace(train_info.artifacts_folder, "")
-
-                input_datetime = task_info["startedAt"]
-                parsed_datetime = datetime.strptime(input_datetime, "%Y-%m-%dT%H:%M:%S.%fZ")
-                date_time = parsed_datetime.strftime("%Y-%m-%d %H:%M:%S")
-
-                experiment_info_data = {
-                    "experiment_name": f"Unknown {self.framework_name} experiment",
-                    "framework_name": self.framework_name,
-                    "model_name": f"Unknown {self.framework_name} model",
-                    "task_type": train_info.task_type,
-                    "project_id": project_id,
-                    "task_id": train_info.task_id,
-                    "model_files": model_files,
-                    "checkpoints": checkpoints,
-                    "best_checkpoint": best_checkpoint,
-                    "artifacts_dir": train_info.artifacts_folder,
-                    "datetime": date_time,
-                }
-
-                experiment_info_fields = {
-                    field.name
-                    for field in ExperimentInfo.__dataclass_fields__.values()  # pylint: disable=no-member
-                }
-                for field in experiment_info_fields:
-                    if field not in experiment_info_data:
-                        experiment_info_data[field] = None
-
-                return ExperimentInfo(**experiment_info_data)
-            except Exception as e:
-                logger.debug(f"Failed to build experiment info: {e}")
-                return None
 
+            input_datetime = task_info["startedAt"]
+            parsed_datetime = datetime.strptime(input_datetime, "%Y-%m-%dT%H:%M:%S.%fZ")
+            date_time = parsed_datetime.strftime("%Y-%m-%d %H:%M:%S")
+
+            experiment_info_data = {
+                "experiment_name": f"Unknown {self.framework_name} experiment",
+                "framework_name": self.framework_name,
+                "model_name": f"Unknown {self.framework_name} model",
+                "task_type": train_info.task_type,
+                "project_id": project_id,
+                "task_id": train_info.task_id,
+                "model_files": model_files,
+                "checkpoints": checkpoints,
+                "best_checkpoint": best_checkpoint,
+                "artifacts_dir": train_info.artifacts_folder,
+                "datetime": date_time,
+            }
+
+            experiment_info_fields = {
+                field.name
+                for field in ExperimentInfo.__dataclass_fields__.values()  # pylint: disable=no-member
+            }
+            for field in experiment_info_fields:
+                if field not in experiment_info_data:
+                    experiment_info_data[field] = None
+            return ExperimentInfo(**experiment_info_data)
+        except Exception as e:
+            logger.debug(f"Failed to build experiment info: {e}")
+            return None
+
+    def get_list_experiment_info(
+        self, sort: Literal["desc", "asc"] = "desc"
+    ) -> List[ExperimentInfo]:
         train_infos = self.get_list(sort)
 
         # Sync version
@@ -607,7 +652,7 @@ class BaseTrainArtifacts:
         with ThreadPoolExecutor() as executor:
             experiment_infos = list(
                 executor.map(
-                    lambda t: build_experiment_info_from_train_info(self._api, t),
+                    lambda t: self.convert_train_to_experiment_info(t),
                     train_infos,
                 )
             )
@@ -621,3 +666,29 @@ class BaseTrainArtifacts:
         :rtype: List[str]
         """
         return self._available_task_types
+
+    def get_info_by_artifacts_dir(
+        self,
+        artifacts_dir: str,
+        return_type: Literal["train_info", "experiment_info"] = "train_info",
+    ) -> Union[TrainInfo, ExperimentInfo, None]:
+        """
+        Get training info by artifacts directory.
+
+        :param artifacts_dir: The artifacts directory.
+        :type artifacts_dir: str
+        :param return_type: The return type, either "train_info" or "experiment_info". Default is "experiment_info".
+        :type return_type: Literal["train_info", "experiment_info"]
+        :return: The training info.
+        :rtype: TrainInfo
+        """
+        for train_info in self.get_list():
+            if train_info.artifacts_folder == artifacts_dir:
+                if return_type == "train_info":
+                    return train_info
+                else:
+                    return self.convert_train_to_experiment_info(train_info)
+
+    # load_custom_checkpoint
+    # inference
+    # fix docstrings :param: x -> :param x:

supervisely/nn/artifacts/detectron2.py

@@ -10,6 +10,11 @@ class Detectron2(BaseTrainArtifacts):
         super().__init__(team_id)
 
         self._app_name = "Train Detectron2"
+        self._slug = "supervisely-ecosystem/detectron2/supervisely/train"
+        self._serve_app_name = "Serve Detectron2"
+        self._serve_slug = (
+            "supervisely-ecosystem/detectron2/supervisely/instance_segmentation/serve"
+        )
         self._framework_name = "Detectron2"
         self._framework_folder = "/detectron2"
         self._weights_folder = "checkpoints"
@@ -19,6 +24,7 @@ class Detectron2(BaseTrainArtifacts):
         self._config_file = "model_config.yaml"
         self._pattern = re_compile(r"^/detectron2/\d+_[^/]+/?$")
         self._available_task_types: List[str] = ["instance segmentation"]
+        self._require_runtime = False
 
     def get_task_id(self, artifacts_folder: str) -> str:
         parts = artifacts_folder.split("/")

supervisely/nn/artifacts/hrda.py

@@ -10,12 +10,16 @@ class HRDA(BaseTrainArtifacts):
         raise NotImplementedError
         # super().__init__(team_id)
         # self._app_name = "Train HRDA"
+        # self._serve_app_name = None
+        # self._slug = None
+        # self._serve_slug = None
         # self._framework_folder = "/HRDA"
         # self._weights_folder = None
         # self._task_type = "semantic segmentation"
         # self._weights_ext = ".pth"
         # self._config_file = "config.py"
         # self._available_task_types: List[str] = ["semantic segmentation"]
+        # self._require_runtime = False
 
     def get_task_id(self, artifacts_folder: str) -> str:
         raise NotImplementedError

supervisely/nn/artifacts/mmclassification.py

@@ -10,6 +10,9 @@ class MMClassification(BaseTrainArtifacts):
         super().__init__(team_id)
 
         self._app_name = "Train MMClassification"
+        self._slug = "supervisely-ecosystem/mmclassification/supervisely/train"
+        self._serve_app_name = "Serve MMClassification"
+        self._serve_slug = "supervisely-ecosystem/mmclassification/supervisely/serve"
         self._framework_name = "MMClassification"
         self._framework_folder = "/mmclassification"
         self._weights_folder = "checkpoints"
@@ -17,6 +20,7 @@ class MMClassification(BaseTrainArtifacts):
         self._weights_ext = ".pth"
         self._pattern = re_compile(r"^/mmclassification/\d+_[^/]+/?$")
         self._available_task_types: List[str] = ["classification"]
+        self._require_runtime = False
 
     def get_task_id(self, artifacts_folder: str) -> str:
         parts = artifacts_folder.split("/")

supervisely/nn/artifacts/mmdetection.py

@@ -13,6 +13,9 @@ class MMDetection(BaseTrainArtifacts):
         super().__init__(team_id)
 
         self._app_name = "Train MMDetection"
+        self._slug = "supervisely-ecosystem/mmdetection/train"
+        self._serve_app_name = "Serve MMDetection"
+        self._serve_slug = "supervisely-ecosystem/mmdetection/serve"
         self._framework_name = "MMDetection"
         self._framework_folder = "/mmdetection"
         self._weights_folder = "checkpoints/data"
@@ -22,6 +25,7 @@ class MMDetection(BaseTrainArtifacts):
         self._config_file = "config.py"
         self._pattern = re_compile(r"^/mmdetection/\d+_[^/]+/?$")
         self._available_task_types: List[str] = ["object detection", "instance segmentation"]
+        self._require_runtime = False
 
     def get_task_id(self, artifacts_folder: str) -> str:
         parts = artifacts_folder.split("/")
@@ -59,6 +63,9 @@ class MMDetection3(BaseTrainArtifacts):
         super().__init__(team_id)
 
         self._app_name = "Train MMDetection 3.0"
+        self._slug = "Serve MMDetection 3.0"
+        self._serve_app_name = "supervisely-ecosystem/train-mmdetection-v3"
+        self._serve_slug = "supervisely-ecosystem/serve-mmdetection-v3"
         self._framework_name = "MMDetection 3.0"
         self._framework_folder = "/mmdetection-3"
         self._weights_folder = None
@@ -67,7 +74,8 @@ class MMDetection3(BaseTrainArtifacts):
         self._config_file = "config.py"
         self._pattern = re_compile(r"^/mmdetection-3/\d+_[^/]+/?$")
         self._available_task_types: List[str] = ["object detection", "instance segmentation"]
-
+        self._require_runtime = False
+        
     def get_task_id(self, artifacts_folder: str) -> str:
         parts = artifacts_folder.split("/")
         if len(parts) < 3:

supervisely/nn/artifacts/mmsegmentation.py

@@ -10,6 +10,9 @@ class MMSegmentation(BaseTrainArtifacts):
         super().__init__(team_id)
 
         self._app_name = "Train MMSegmentation"
+        self._slug = "supervisely-ecosystem/mmsegmentation/train"
+        self._serve_app_name = "Serve MMSegmentation"
+        self._serve_slug = "supervisely-ecosystem/mmsegmentation/serve"
         self._framework_name = "MMSegmentation"
         self._framework_folder = "/mmsegmentation"
         self._weights_folder = "checkpoints/data"
@@ -18,6 +21,7 @@ class MMSegmentation(BaseTrainArtifacts):
         self._config_file = "config.py"
         self._pattern = re_compile(r"^/mmsegmentation/\d+_[^/]+/?$")
         self._available_task_types: List[str] = ["instance segmentation"]
+        self._require_runtime = False
 
     def get_task_id(self, artifacts_folder: str) -> str:
         return artifacts_folder.split("/")[2].split("_")[0]

supervisely/nn/artifacts/ritm.py

@@ -10,6 +10,9 @@ class RITM(BaseTrainArtifacts):
         super().__init__(team_id)
 
         self._app_name = "Train RITM"
+        self._slug = "supervisely-ecosystem/ritm-training/supervisely/train"
+        self._serve_app_name = None
+        self._serve_slug = None
         self._framework_name = "RITM"
         self._framework_folder = "/RITM_training"
         self._weights_folder = "checkpoints"
@@ -18,6 +21,7 @@ class RITM(BaseTrainArtifacts):
         self._weights_ext = ".pth"
         self._pattern = re_compile(r"^/RITM_training/\d+_[^/]+/?$")
         self._available_task_types: List[str] = ["interactive segmentation"]
+        self._require_runtime = False
 
     def get_task_id(self, artifacts_folder: str) -> str:
         parts = artifacts_folder.split("/")

supervisely/nn/artifacts/rtdetr.py

@@ -10,6 +10,9 @@ class RTDETR(BaseTrainArtifacts):
         super().__init__(team_id)
 
         self._app_name = "Train RT-DETR"
+        self._slug = "supervisely-ecosystem/rt-detr/supervisely_integration/train"
+        self._serve_app_name = "Serve RT-DETR"
+        self._serve_slug = "supervisely-ecosystem/rt-detr/supervisely_integration/serve"
         self._framework_name = "RT-DETR"
         self._framework_folder = "/RT-DETR"
         self._weights_folder = "weights"
@@ -18,6 +21,7 @@ class RTDETR(BaseTrainArtifacts):
         self._config_file = "config.yml"
         self._pattern = re_compile(r"^/RT-DETR/[^/]+/\d+/?$")
         self._available_task_types: List[str] = ["object detection"]
+        self._require_runtime = False
 
     def get_task_id(self, artifacts_folder: str) -> str:
         return artifacts_folder.split("/")[-1]

supervisely/nn/artifacts/unet.py

@@ -10,6 +10,9 @@ class UNet(BaseTrainArtifacts):
         super().__init__(team_id)
 
         self._app_name = "Train UNet"
+        self._slug = "supervisely-ecosystem/unet/supervisely/train"
+        self._serve_app_name = "Serve UNet"
+        self._serve_slug = "supervisely-ecosystem/unet/supervisely/serve"
         self._framework_name = "UNet"
         self._framework_folder = "/unet"
         self._weights_folder = "checkpoints"
@@ -18,6 +21,7 @@ class UNet(BaseTrainArtifacts):
         self._config_file = "train_args.json"
         self._pattern = re_compile(r"^/unet/\d+_[^/]+/?$")
         self._available_task_types: List[str] = ["semantic segmentation"]
+        self._require_runtime = False
 
     def get_task_id(self, artifacts_folder: str) -> str:
         parts = artifacts_folder.split("/")