supervisely 6.73.422__py3-none-any.whl → 6.73.424__py3-none-any.whl

supervisely/nn/training/gui/gui.py

@@ -6,6 +6,7 @@ training workflows in Supervisely.
 """
 
 import os
+import json
 from os import environ, getenv
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
@@ -846,16 +847,14 @@ class TrainGUI:
             }
         """
         if isinstance(app_state, str):
-            app_state = sly_json.load_json_file(app_state)
+            if os.path.isfile(app_state):
+                app_state = sly_json.load_json_file(app_state)
+            else:
+                app_state = json.loads(app_state)
 
         app_state = self.validate_app_state(app_state)
         options = app_state.get("options", {})
 
-        # Set experiment name
-        experiment_name = app_state.get("experiment_name")
-        if experiment_name is not None:
-            self.training_process.set_experiment_name(experiment_name)
-
         # Run init-steps and stop on validation failure
         def _run_step(init_fn, settings) -> bool:
             if not init_fn(settings, options, click_cb, validate_steps):
@@ -881,6 +880,12 @@ class TrainGUI:
                 logger.info(
                     f"Step '{step_name}' {idx}/{len(_steps)} has been validated successfully"
                 )
+
+        # Set experiment name
+        experiment_name = app_state.get("experiment_name")
+        if experiment_name is not None and experiment_name != "":
+            self.training_process.set_experiment_name(experiment_name)
+
         if validate_steps:
             logger.info(f"All steps have been validated successfully")
         # ------------------------------------------------------------------ #
@@ -990,6 +995,13 @@ class TrainGUI:
             self.classes_selector.convert_class_shapes_checkbox.check()
 
         # Set Classes
+        if all(isinstance(c, int) for c in classes_settings):
+            project_classes = []
+            for obj_class in self.project_meta.obj_classes:
+                if obj_class.sly_id in classes_settings:
+                    project_classes.append(obj_class.name)
+            classes_settings = project_classes
+
         self.classes_selector.set_classes(classes_settings)
         is_valid = True
         if validate:

supervisely/nn/training/train_app.py

@@ -1096,6 +1096,7 @@ class TrainApp:
                 project_id=self.project_id,
                 dest_dir=self.project_dir,
                 dataset_ids=[ds_info.id for ds_info in dataset_infos],
+                save_image_info=True,
                 log_progress=True,
                 progress_cb=pbar.update,
             )

{supervisely-6.73.422.dist-info → supervisely-6.73.424.dist-info}/METADATA

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

{supervisely-6.73.422.dist-info → supervisely-6.73.424.dist-info}/RECORD

@@ -1006,10 +1006,10 @@ supervisely/nn/tracker/botsort/tracker/kalman_filter.py,sha256=waTArMcbmpHAzb57a
 supervisely/nn/tracker/botsort/tracker/matching.py,sha256=bgnheHwWD3XZSI3OJVfdrU5bYJ44rxPHzzSElfg6LZM,6600
 supervisely/nn/tracker/botsort/tracker/mc_bot_sort.py,sha256=AE_yi4pVg_bIsK-6-ZlPziqb98tA-9iWgjT8aRE-8sA,17775
 supervisely/nn/training/__init__.py,sha256=gY4PCykJ-42MWKsqb9kl-skemKa8yB6t_fb5kzqR66U,111
-supervisely/nn/training/train_app.py,sha256=dSvXDLpXs726uiCHjQy7KW6XTuwdAapaM720mKsQeNU,131899
+supervisely/nn/training/train_app.py,sha256=DW9J6qZEebRFdq59sexvEZTWyExisbz7Z2lHXwEkPjY,131937
 supervisely/nn/training/gui/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 supervisely/nn/training/gui/classes_selector.py,sha256=tqmVwUfC2u5K53mZmvDvNOhu9Mw5mddjpB2kxRXXUO8,12453
-supervisely/nn/training/gui/gui.py,sha256=uwl6g5ytgs0EcBJDBL3MHBqVVdfZDyuUdybfU2SnjZE,51128
+supervisely/nn/training/gui/gui.py,sha256=Yjavju7HbgYZMZvhI_kdrz65Uf87Qc4-KR7SVkfUXTc,51600
 supervisely/nn/training/gui/hyperparameters_selector.py,sha256=tEyppV5ay7nECi6qBQ9seKgfidPa_FxxtGmqsfQapa4,7768
 supervisely/nn/training/gui/input_selector.py,sha256=rmirJzpdxuYONI6y5_cvMdGWBJ--T20YTsISghATHu4,2510
 supervisely/nn/training/gui/model_selector.py,sha256=YKBAk6MheulFEl9TF9_mVtE3-Hsc0B3LmeOzMiV6AlQ,7487
@@ -1121,9 +1121,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.422.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.422.dist-info/METADATA,sha256=5ieuqnn6OCtRmk3I-RqDMFPT4uy7BQh3ziMSlRQK25M,35433
-supervisely-6.73.422.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-supervisely-6.73.422.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.422.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.422.dist-info/RECORD,,
+supervisely-6.73.424.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.424.dist-info/METADATA,sha256=wi_URnVWvi7XZgZnKHeNriZGnIauWTCXPW-MoWqtrjU,35433
+supervisely-6.73.424.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+supervisely-6.73.424.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.424.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.424.dist-info/RECORD,,