supervisely 6.73.390__py3-none-any.whl → 6.73.392__py3-none-any.whl

supervisely/nn/training/gui/hyperparameters_selector.py

@@ -14,7 +14,7 @@ from supervisely.app.widgets import (
 class HyperparametersSelector:
     title = "Hyperparameters"
     description = "Set hyperparameters for training"
-    lock_message = "Select model to unlock"
+    lock_message = "Select previous step to unlock"
 
     def __init__(self, hyperparameters: dict, app_options: dict = {}):
         # Init widgets
@@ -60,7 +60,7 @@ class HyperparametersSelector:
                 f"Learn more about Model Benchmark in the {docs_link}.", status="info"
             )
             self.model_benchmark_auto_convert_warning = Text(
-                text="Project will be automatically converted according to CV task for Model Evaluation.",
+                text="Project will be automatically converted according to CV task and uploaded for Model Evaluation.",
                 status="warning",
             )
             self.model_benchmark_auto_convert_warning.hide()

supervisely/nn/training/gui/model_selector.py

@@ -1,6 +1,7 @@
 from typing import Literal
 
 import supervisely.io.env as sly_env
+import supervisely.io.fs as sly_fs
 from supervisely import logger
 from supervisely.api.api import Api
 from supervisely.app.widgets import (
@@ -15,12 +16,13 @@ from supervisely.app.widgets import (
 from supervisely.nn.artifacts.utils import FrameworkMapper
 from supervisely.nn.experiments import get_experiment_infos
 from supervisely.nn.utils import ModelSource, _get_model_name
+from supervisely._utils import get_filename_from_headers
 
 
 class ModelSelector:
     title = "Select Model"
     description = "Select a model for training"
-    lock_message = "Select classes to unlock"
+    lock_message = "Select previous step to unlock"
 
     def __init__(self, api: Api, framework: str, models: list, app_options: dict = {}):
         # Init widgets
@@ -125,6 +127,45 @@ class ModelSelector:
         else:
             return self.experiment_selector.get_selected_experiment_info()
 
+    def get_checkpoint_name(self) -> str:
+        if self.get_model_source() == ModelSource.PRETRAINED:
+            selected_row = self.pretrained_models_table.get_selected_row()
+            meta = selected_row.get("meta", None)
+            if meta is not None:
+                model_files = meta.get("model_files", None)
+                if model_files is not None:
+                    checkpoint_name = model_files.get("checkpoint", None)
+                    if checkpoint_name is not None:
+                        is_url = sly_fs.str_is_url(checkpoint_name)
+                        if (not is_url) or checkpoint_name.lower().endswith((".pth", ".pt")):
+                            checkpoint_name = sly_fs.get_file_name_with_ext(checkpoint_name)
+                        else:
+                            checkpoint_name = get_filename_from_headers(checkpoint_name)
+                        return checkpoint_name
+            return None
+        else:
+            checkpoint_name = self.experiment_selector.get_selected_checkpoint_name()
+        return checkpoint_name
+    
+    def get_checkpoint_link(self) -> str:
+        if self.get_model_source() == ModelSource.PRETRAINED:
+            selected_row = self.pretrained_models_table.get_selected_row()
+            meta = selected_row.get("meta", None)
+            if meta is not None:
+                model_files = meta.get("model_files", None)
+                if model_files is not None:
+                    checkpoint_link = model_files.get("checkpoint", None)
+                    if checkpoint_link is not None:
+                        is_url = sly_fs.str_is_url(checkpoint_link)
+                        if checkpoint_link.startswith("/experiments/"):
+                            is_url = True
+                        if is_url:
+                            return checkpoint_link
+            return None
+        else:
+            checkpoint_link = self.experiment_selector.get_selected_checkpoint_path()
+        return checkpoint_link
+
     def validate_step(self) -> bool:
         self.validator_text.hide()
         model_info = self.get_model_info()

supervisely/nn/training/gui/tags_selector.py

@@ -5,7 +5,7 @@ from supervisely.app.widgets import Button, Card, Container, TagsTable, Text
 class TagsSelector:
     title = "Tags Selector"
     description = "Select tags that will be used for training"
-    lock_message = "Select training and validation splits to unlock"
+    lock_message = "Select previous step to unlock"
 
     def __init__(self, project_id: int, tags: list, app_options: dict = {}):
         # Init widgets

supervisely/nn/training/gui/train_val_splits_selector.py

@@ -2,12 +2,12 @@ from typing import List
 
 from supervisely import Api, Project
 from supervisely.app.widgets import Button, Card, Container, Text, TrainValSplits
-
+from supervisely.api.module_api import ApiField
 
 class TrainValSplitsSelector:
     title = "Train / Val Splits"
     description = "Select train and val splits for training"
-    lock_message = "Select input options to unlock"
+    lock_message = "Select previous step to unlock"
 
     def __init__(self, api: Api, project_id: int, app_options: dict = {}):
         # Init widgets
@@ -222,12 +222,13 @@ class TrainValSplitsSelector:
                 return False
 
             # Check if datasets are not empty
-            stats = self.api.project.get_stats(self.project_id)
+            filters = [{ ApiField.FIELD: ApiField.ID, ApiField.OPERATOR: "in", ApiField.VALUE: train_dataset_id + val_dataset_id}]
+            selected_datasets = self.api.dataset.get_list(self.project_id, filters, recursive=True)
             datasets_count = {}
-            for dataset in stats["images"]["datasets"]:
-                datasets_count[dataset["id"]] = {
-                    "name": dataset["name"],
-                    "total": dataset["imagesInDataset"],
+            for dataset in selected_datasets:
+                datasets_count[dataset.id] = {
+                    "name": dataset.name,
+                    "total": dataset.images_count,
                 }
 
             empty_dataset_names = []

supervisely/nn/training/gui/training_artifacts.py

@@ -64,7 +64,16 @@ class TrainingArtifacts:
         self.display_widgets.extend([self.validator_text])
 
         # Outputs
-        self.artifacts_thumbnail = FolderThumbnail()
+        need_generate_report = self.app_options.get("generate_report", True)
+        if need_generate_report:
+            self.artifacts_thumbnail = ReportThumbnail(
+                title="Experiment Report",
+                color="#5fa8ff",
+                bg_color="#e6f3ff",
+                report_type="experiment",
+            )
+        else:
+            self.artifacts_thumbnail = FolderThumbnail()
         self.artifacts_thumbnail.hide()
 
         self.artifacts_field = Field(

supervisely/nn/training/gui/training_process.py

@@ -1,5 +1,11 @@
 from typing import Any, Dict
 
+# Safe optional import for torch to prevent pylint import-error when the library is absent.
+try:
+    import torch  # type: ignore
+except ImportError:  # pragma: no cover
+    torch = None  # type: ignore
+
 from supervisely.app.widgets import (
     Button,
     Card,
@@ -15,7 +21,7 @@ from supervisely.app.widgets import (
 class TrainingProcess:
     title = "Training Process"
     description = "Manage training process"
-    lock_message = "Select hyperparameters to unlock"
+    lock_message = "Select previous step to unlock"
 
     def __init__(self, app_options: Dict[str, Any]):
         # Initialize widgets to None
@@ -93,6 +99,16 @@ class TrainingProcess:
         else:
             return "cuda:0"
 
+    def get_device_name(self) -> str:
+        device = self.get_device()
+
+        if torch is not None and device.startswith("cuda"):
+            device_name = torch.cuda.get_device_name(device)
+        else:
+            device_name = "CPU"
+
+        return device_name
+
     def get_experiment_name(self) -> str:
         return self.experiment_name_input.get_value()