supervisely 6.73.260__py3-none-any.whl → 6.73.261__py3-none-any.whl

supervisely/app/widgets/experiment_selector/experiment_selector.py

@@ -19,6 +19,8 @@ from supervisely.app.widgets import (
 )
 from supervisely.io.fs import get_file_name_with_ext
 from supervisely.nn.experiments import ExperimentInfo
+from supervisely.nn.utils import ModelSource
+
 
 WEIGHTS_DIR = "weights"
 
@@ -473,6 +475,17 @@ class ExperimentSelector(Widget):
         full_model_files["checkpoint"] = self.get_selected_checkpoint_path()
         return full_model_files
 
+    def get_deploy_params(self) -> Dict[str, Any]:
+        """
+        Returns a dictionary with deploy parameters except runtime and device keys.
+        """
+        deploy_params = {
+            "model_source": ModelSource.CUSTOM,
+            "model_files": self.get_model_files(),
+            "model_info": self.get_selected_experiment_info(),
+        }
+        return deploy_params
+
     def set_active_row(self, row_index: int) -> None:
         if row_index < 0 or row_index > len(self._rows) - 1:
             raise ValueError(f'Row with index "{row_index}" does not exist')

supervisely/app/widgets/pretrained_models_selector/pretrained_models_selector.py

@@ -1,9 +1,10 @@
-from typing import Dict, List, Union
+from typing import Dict, List, Literal, Union
 
 from supervisely.api.api import Api
 from supervisely.app.content import DataJson, StateJson
 from supervisely.app.widgets import Widget
 from supervisely.io.fs import get_file_ext
+from supervisely.nn.utils import ModelSource
 
 
 class PretrainedModelsSelector(Widget):
@@ -134,39 +135,71 @@ class PretrainedModelsSelector(Widget):
             selected_row_index = int(widget_actual_state["selectedRow"])
             return models[selected_row_index]
 
-    def get_selected_model_params(self, model_name_column: str = "Model") -> Union[Dict, None]:
+    def get_selected_model_params(
+        self,
+        model_name_column: str = "Model",
+        train_version: Literal["v1", "v2"] = "v1",
+    ) -> Union[Dict, None]:
         selected_model = self.get_selected_row()
         if selected_model is None:
             return {}
 
-        model_name = selected_model.get(model_name_column)
-        if model_name is None:
-            raise ValueError(
-                "Could not find model name. Make sure you have column 'Model' in your models list."
-            )
-        checkpoint_url = selected_model.get("meta", {}).get("weights_url")
-        if checkpoint_url is None:
-            pass
-
-        checkpoint_ext = get_file_ext(checkpoint_url)
-        checkpoint_name = f"{model_name.lower()}{checkpoint_ext}"
-
-        task_type = self.get_selected_task_type()
-        model_params = {
-            "model_source": "Pretrained models",
-            "task_type": task_type,
-            "checkpoint_name": checkpoint_name,
-            "checkpoint_url": checkpoint_url,
-        }
-
-        if len(self._arch_types) > 1:
-            arch_type = self.get_selected_arch_type()
-            model_params["arch_type"] = arch_type
-
-        config_url = selected_model.get("meta", {}).get("config_url")
-        if config_url is not None:
-            model_params["config_url"] = config_url
+        if train_version == "v1":
+            model_name = selected_model.get(model_name_column)
+            if model_name is None:
+                raise ValueError(
+                    "Could not find model name. Make sure you have column 'Model' in your models list."
+                )
+
+            model_meta = selected_model.get("meta")
+            if model_meta is None:
+                raise ValueError(
+                    "Could not find model meta. Make sure you have key 'meta' in your models configuration list."
+                )
+            checkpoint_url = model_meta.get("weights_url")
+            if checkpoint_url is None:
+                model_files = model_meta.get("model_files")
+                if model_files is None:
+                    raise ValueError(
+                        "Could not find model files. Make sure you have key 'model_files' or 'weights_url' in 'meta' in your models configuration list."
+                    )
+                checkpoint_url = model_files.get("checkpoint")
+                if checkpoint_url is None:
+                    raise ValueError(
+                        "Could not find checkpoint url. Make sure you have key 'checkpoint' in 'model_files' in 'meta' in your models configuration list."
+                    )
+
+            checkpoint_ext = get_file_ext(checkpoint_url)
+            checkpoint_name = f"{model_name.lower()}{checkpoint_ext}"
+
+            task_type = self.get_selected_task_type()
+            model_params = {
+                "model_source": "Pretrained models",
+                "task_type": task_type,
+                "checkpoint_name": checkpoint_name,
+                "checkpoint_url": checkpoint_url,
+            }
 
+            if len(self._arch_types) > 1:
+                arch_type = self.get_selected_arch_type()
+                model_params["arch_type"] = arch_type
+
+            config_url = selected_model.get("meta", {}).get("config_url")
+            if config_url is not None:
+                model_params["config_url"] = config_url
+        elif train_version == "v2":
+            model_info = self.get_selected_row()
+            meta = model_info.get("meta")
+            if meta is None:
+                raise ValueError("key 'meta' not found in model configuration")
+            model_files = meta.get("model_files")
+            if model_files is None:
+                raise ValueError("key 'model_files' not found in key 'meta' in model configuration")
+            model_params = {
+                "model_source": ModelSource.PRETRAINED,
+                "model_info": model_info,
+                "model_files": model_files,
+            }
         return model_params
 
     def get_selected_row_index(self, state=StateJson()) -> Union[int, None]:

supervisely/nn/artifacts/artifacts.py

@@ -536,58 +536,62 @@ class BaseTrainArtifacts:
             api: Api, train_info: TrainInfo
         ) -> ExperimentInfo:
 
-            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 = 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, ""
+            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
                 )
-
-            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)
+                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.warning(f"Failed to build experiment info: {e}")
+                return None
 
         train_infos = self.get_list(sort)
 

supervisely/nn/inference/inference.py

@@ -571,7 +571,9 @@ class Inference:
                     ) as download_pbar:
                         self.gui.download_progress.show()
                         sly_fs.download(
-                            url=file_url, save_path=file_path, progress=download_pbar.update
+                            url=file_url,
+                            save_path=file_path,
+                            progress=download_pbar.update,
                         )
                     local_model_files[file] = file_path
             else:
@@ -590,6 +592,10 @@ class Inference:
         for file in model_files:
             file_url = model_files[file]
             file_info = self.api.file.get_info_by_path(team_id, file_url)
+            if file_info is None:
+                raise FileNotFoundError(
+                    f"File '{file_url}' not found in Team Files. Make sure the file exists."
+                )
             file_size = file_info.sizeb
             file_name = os.path.basename(file_url)
             file_path = os.path.join(self.model_dir, file_name)
@@ -1374,7 +1380,11 @@ class Inference:
         return result
 
     def _inference_images_ids(
-        self, api: Api, state: dict, images_ids: List[int], async_inference_request_uuid: str = None
+        self,
+        api: Api,
+        state: dict,
+        images_ids: List[int],
+        async_inference_request_uuid: str = None,
     ):
         """Inference images by ids.
         If "output_project_id" in state, upload images and annotations to the output project.
@@ -1799,7 +1809,8 @@ class Inference:
             dataset_id = _get_or_create_new_dataset(output_project_id, src_dataset_id)
             image_names = [result["image_name"] for result in results]
             image_infos = api.image.get_list(
-                dataset_id, filters=[{"field": "name", "operator": "in", "value": image_names}]
+                dataset_id,
+                filters=[{"field": "name", "operator": "in", "value": image_names}],
             )
             meta_changed = False
             anns = []
@@ -2104,7 +2115,10 @@ class Inference:
                 # Read images
                 if cache_project_on_model:
                     images_nps = []
-                    for dataset_id, images_infos in images_infos_batch_by_dataset.items():
+                    for (
+                        dataset_id,
+                        images_infos,
+                    ) in images_infos_batch_by_dataset.items():
                         dataset_info = datasets_infos_dict[dataset_id]
                         images_paths, _ = zip(
                             *read_from_cached_project(
@@ -2116,7 +2130,10 @@ class Inference:
                         images_nps.extend([sly_image.read(path) for path in images_paths])
                 else:
                     images_nps = []
-                    for dataset_id, images_infos in images_infos_batch_by_dataset.items():
+                    for (
+                        dataset_id,
+                        images_infos,
+                    ) in images_infos_batch_by_dataset.items():
                         images_nps.extend(
                             self.cache.download_images(
                                 api,
@@ -2845,7 +2862,8 @@ def update_meta_and_ann(meta: ProjectMeta, ann: Annotation):
     """Update project meta and annotation to match each other
     If obj class or tag meta from annotation conflicts with project meta
     add suffix to obj class or tag meta.
-    Return tuple of updated project meta, annotation and boolean flag if meta was changed."""
+    Return tuple of updated project meta, annotation and boolean flag if meta was changed.
+    """
     obj_classes_suffixes = ["_nn"]
     tag_meta_suffixes = ["_nn"]
     ann_obj_classes = {}

{supervisely-6.73.260.dist-info → supervisely-6.73.261.dist-info}/METADATA

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

{supervisely-6.73.260.dist-info → supervisely-6.73.261.dist-info}/RECORD

@@ -250,7 +250,7 @@ supervisely/app/widgets/empty/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5N
 supervisely/app/widgets/empty/empty.py,sha256=fCr8I7CQ2XLo59bl2txjDrblOGiu0TzUcM-Pq6s7gKY,1285
 supervisely/app/widgets/empty/template.html,sha256=aDBKkin5aLuqByzNN517-rTYCGIg5SPKgnysYMPYjv8,40
 supervisely/app/widgets/experiment_selector/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-supervisely/app/widgets/experiment_selector/experiment_selector.py,sha256=XZxhpD3utmHsHhKxo34-wJIfOaKXXznrFQVnVaHVWik,19456
+supervisely/app/widgets/experiment_selector/experiment_selector.py,sha256=D1RrMuA6xXg8M5DyBUKro-AUqj3oFNMIlo6pZUR0SQY,19889
 supervisely/app/widgets/experiment_selector/style.css,sha256=-zPPXHnJvatYj_xVVAb7T8uoSsUTyhm5xCKWkkFQ78E,548
 supervisely/app/widgets/experiment_selector/template.html,sha256=k7f_Xl6nDUXXwu6IY_RblYni5TbZRRxCBduY5O_SyFs,2908
 supervisely/app/widgets/fast_table/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -386,7 +386,7 @@ supervisely/app/widgets/pagination/template.html,sha256=1z9pt2SOTjA5Kmt8YjSiyO8X
 supervisely/app/widgets/pie_chart/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 supervisely/app/widgets/pie_chart/pie_chart.py,sha256=E0erw7kPXiudJzNhWMKdOnMgK9u6yYlPw10noKoO2jw,7809
 supervisely/app/widgets/pretrained_models_selector/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-supervisely/app/widgets/pretrained_models_selector/pretrained_models_selector.py,sha256=h6R5bFa1BNCqskFKV9yMGzhY4rf1ZT3ZBh8MKfG87Ko,12394
+supervisely/app/widgets/pretrained_models_selector/pretrained_models_selector.py,sha256=y21fRcDWuP1RbqxNzy3MRxWFi8-f9pV0SsDaNIWqlvE,14102
 supervisely/app/widgets/pretrained_models_selector/style.css,sha256=po3FssuZhg3lKFU3VcTLqTW-qTCXLDnxYi2lCtYXhBc,363
 supervisely/app/widgets/pretrained_models_selector/template.html,sha256=4owO7X84Ii35vMioeTOwMAdR9OwlvLFe19MNPDX4kWM,5170
 supervisely/app/widgets/project_selector/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -718,7 +718,7 @@ supervisely/nn/prediction_dto.py,sha256=8QQE6h_feOf3bjWtyG_PoU8FIQrr4g8PoMOyoscm
 supervisely/nn/task_type.py,sha256=UJvSJ4L3I08j_e6sU6Ptu7kS5p1H09rfhfoDUSZ2iys,522
 supervisely/nn/utils.py,sha256=-Xjv5KLu8CTtyi7acqsIX1E0dDwKZPED4D6b4Z_Ln3k,1451
 supervisely/nn/artifacts/__init__.py,sha256=m7KYTMzEJnoV9wcU_0xzgLuPz69Dqp9va0fP32tohV4,576
-supervisely/nn/artifacts/artifacts.py,sha256=SWVwaE40SPNElLp2NsRSOVh_ESRIXXZ1I-tMLEUYmTQ,20751
+supervisely/nn/artifacts/artifacts.py,sha256=mizJzrzAMoURYnHU-Ap8OdUAxRngDtR5PC0RyDxBp8k,21079
 supervisely/nn/artifacts/detectron2.py,sha256=6iu5Yslc-SxCKJVNl6sn84qeXmD-JIQShJIxuLdzf2o,1673
 supervisely/nn/artifacts/hrda.py,sha256=3DzbjDIt9YuLozMrKmYYw13oxc14xju2vzbxKVq2G4I,1073
 supervisely/nn/artifacts/mmclassification.py,sha256=M0m9HHF5koHsl5RuFkRU0_clckA1sFty3X6awB2eKNo,1527
@@ -859,7 +859,7 @@ supervisely/nn/benchmark/visualization/widgets/table/__init__.py,sha256=47DEQpj8
 supervisely/nn/benchmark/visualization/widgets/table/table.py,sha256=atmDnF1Af6qLQBUjLhK18RMDKAYlxnsuVHMSEa5a-e8,4319
 supervisely/nn/inference/__init__.py,sha256=mtEci4Puu-fRXDnGn8RP47o97rv3VTE0hjbYO34Zwqg,1622
 supervisely/nn/inference/cache.py,sha256=KvzCgMbEBLdiJAxJDLicIPKAlYb52P9_kpNPWfiVY8Y,28194
-supervisely/nn/inference/inference.py,sha256=1QLeQmAmo29LoM9uFQjI5iDcuOyoltL4QR2Han9f2MM,128420
+supervisely/nn/inference/inference.py,sha256=acNJH-1UthvJgyhI-9KJv8hzo8LjOVVdXPkoEuwfRLg,128864
 supervisely/nn/inference/session.py,sha256=jmkkxbe2kH-lEgUU6Afh62jP68dxfhF5v6OGDfLU62E,35757
 supervisely/nn/inference/video_inference.py,sha256=8Bshjr6rDyLay5Za8IB8Dr6FURMO2R_v7aELasO8pR4,5746
 supervisely/nn/inference/gui/__init__.py,sha256=wCxd-lF5Zhcwsis-wScDA8n1Gk_1O00PKgDviUZ3F1U,221
@@ -1057,9 +1057,9 @@ supervisely/worker_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 supervisely/worker_proto/worker_api_pb2.py,sha256=VQfi5JRBHs2pFCK1snec3JECgGnua3Xjqw_-b3aFxuM,59142
 supervisely/worker_proto/worker_api_pb2_grpc.py,sha256=3BwQXOaP9qpdi0Dt9EKG--Lm8KGN0C5AgmUfRv77_Jk,28940
 supervisely_lib/__init__.py,sha256=7-3QnN8Zf0wj8NCr2oJmqoQWMKKPKTECvjH9pd2S5vY,159
-supervisely-6.73.260.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.260.dist-info/METADATA,sha256=CPwcqKvIAjMyeGM6DRLQKHF3owWEcixltIqVe4T0x-E,33573
-supervisely-6.73.260.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-supervisely-6.73.260.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.260.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.260.dist-info/RECORD,,
+supervisely-6.73.261.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.261.dist-info/METADATA,sha256=-ebDT49QDbB4Ly0fCqehbeu-_SN7bLV1lObohSxHS4M,33573
+supervisely-6.73.261.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+supervisely-6.73.261.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.261.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.261.dist-info/RECORD,,