supervisely 6.73.339__py3-none-any.whl → 6.73.341__py3-none-any.whl

supervisely/api/image_api.py

@@ -2245,7 +2245,7 @@ class ImageApi(RemoveableBulkModuleApi):
         if len(ids) == 0:
             return
 
-        existing_images = self.get_list(dst_dataset_id)
+        existing_images = self.get_list(dst_dataset_id, force_metadata_for_links=False)
         existing_names = {image.name for image in existing_images}
 
         ids_info = self.get_info_by_id_batch(ids, force_metadata_for_links=False)
@@ -2271,7 +2271,8 @@ class ImageApi(RemoveableBulkModuleApi):
                     "names intersection"
                 )
 
-        new_images = self.upload_ids(dst_dataset_id, new_names, ids, progress_cb)
+        img_metas = [info.meta or {} for info in ids_info]
+        new_images = self.upload_ids(dst_dataset_id, new_names, ids, progress_cb, metas=img_metas)
         new_ids = [new_image.id for new_image in new_images]
 
         if with_annotations:
@@ -2356,12 +2357,14 @@ class ImageApi(RemoveableBulkModuleApi):
                 raise RuntimeError("len(dst_names) != len(src_image_infos)")
             new_names = dst_names
 
+        img_metas = [info.meta or {} for info in src_image_infos]
         src_ids = [info.id for info in src_image_infos]
         new_images = self.upload_ids(
             dst_dataset_id,
             new_names,
             src_ids,
             progress_cb,
+            metas=img_metas,
             batch_size=batch_size,
             force_metadata_for_links=False,
             infos=src_image_infos,

supervisely/nn/inference/inference.py

@@ -598,15 +598,16 @@ class Inference:
     def _download_model_files(self, deploy_params: dict, log_progress: bool = True) -> dict:
         if deploy_params["runtime"] != RuntimeType.PYTORCH:
             export = deploy_params["model_info"].get("export", {})
-            export_model = export.get(deploy_params["runtime"], None)
-            if export_model is not None:
-                if sly_fs.get_file_name(export_model) == sly_fs.get_file_name(
-                    deploy_params["model_files"]["checkpoint"]
-                ):
-                    deploy_params["model_files"]["checkpoint"] = (
-                        deploy_params["model_info"]["artifacts_dir"] + export_model
-                    )
-                    logger.info(f"Found model checkpoint for '{deploy_params['runtime']}'")
+            if export is not None:
+                export_model = export.get(deploy_params["runtime"], None)
+                if export_model is not None:
+                    if sly_fs.get_file_name(export_model) == sly_fs.get_file_name(
+                        deploy_params["model_files"]["checkpoint"]
+                    ):
+                        deploy_params["model_files"]["checkpoint"] = (
+                            deploy_params["model_info"]["artifacts_dir"] + export_model
+                        )
+                        logger.info(f"Found model checkpoint for '{deploy_params['runtime']}'")
 
         if deploy_params["model_source"] == ModelSource.PRETRAINED:
             return self._download_pretrained_model(deploy_params["model_files"], log_progress)

supervisely/nn/training/train_app.py

@@ -2524,17 +2524,26 @@ class TrainApp:
             self.gui.hyperparameters_selector.get_export_onnx_checkbox_value() is True
             and self._convert_onnx_func is not None
         ):
-            self._set_text_status("export_onnx")
-            onnx_path = self._convert_onnx_func(experiment_info)
-            export_weights[RuntimeType.ONNXRUNTIME] = onnx_path
+            try:
+                self._set_text_status("export_onnx")
+                onnx_path = self._convert_onnx_func(experiment_info)
+                if sly_fs.file_exists(onnx_path):
+                    export_weights[RuntimeType.ONNXRUNTIME] = onnx_path
+            except Exception as e:
+                logger.error(f"Failed to export ONNX model: {e}")
 
         if (
             self.gui.hyperparameters_selector.get_export_tensorrt_checkbox_value() is True
             and self._convert_tensorrt_func is not None
         ):
-            self._set_text_status("export_trt")
-            tensorrt_path = self._convert_tensorrt_func(experiment_info)
-            export_weights[RuntimeType.TENSORRT] = tensorrt_path
+            try:
+                self._set_text_status("export_trt")
+                tensorrt_path = self._convert_tensorrt_func(experiment_info)
+                if sly_fs.file_exists(tensorrt_path):
+                    export_weights[RuntimeType.TENSORRT] = tensorrt_path
+            except Exception as e:
+                logger.error(f"Failed to export TensorRT model: {e}")
+                
         return export_weights
 
     def _upload_export_weights(

{supervisely-6.73.339.dist-info → supervisely-6.73.341.dist-info}/METADATA

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

{supervisely-6.73.339.dist-info → supervisely-6.73.341.dist-info}/RECORD

@@ -28,7 +28,7 @@ supervisely/api/dataset_api.py,sha256=GH7prDRJKyJlTv_7_Y-RkTwJN7ED4EkXNqqmi3iIdI
 supervisely/api/file_api.py,sha256=bVWv6kf3B5n6qlB14HmUa6iUr8ara5cr-pPK8QC7XWg,92932
 supervisely/api/github_api.py,sha256=NIexNjEer9H5rf5sw2LEZd7C1WR-tK4t6IZzsgeAAwQ,623
 supervisely/api/image_annotation_tool_api.py,sha256=YcUo78jRDBJYvIjrd-Y6FJAasLta54nnxhyaGyanovA,5237
-supervisely/api/image_api.py,sha256=WIML_6N1qgOWBm3acexmGSWz4hAaSxlYmUtbytROaP8,192375
+supervisely/api/image_api.py,sha256=MIs4hVOy0J7o7KHF2aHt3V9noin5Lr2q4HGOIz5ONMw,192578
 supervisely/api/import_storage_api.py,sha256=BDCgmR0Hv6OoiRHLCVPKt3iDxSVlQp1WrnKhAK_Zl84,460
 supervisely/api/issues_api.py,sha256=BqDJXmNoTzwc3xe6_-mA7FDFC5QQ-ahGbXk_HmpkSeQ,17925
 supervisely/api/labeling_job_api.py,sha256=VM1pjM65cUTeer1hrI7cSmCUOHJb7fzK6gJ-abA07hg,55097
@@ -883,7 +883,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=QFukX2ip-U7263aEPCF_UCFwj6EujbMnsgrXp5Bbt8I,1623
 supervisely/nn/inference/cache.py,sha256=LAirR5mFHCtK59EO1lefQ2qhpp0vBvRTH26EVrs13Y0,32073
-supervisely/nn/inference/inference.py,sha256=dzfvHjisVYgsijnHYazb_y0mXYZQJX0E1fEsDY6S5lw,169730
+supervisely/nn/inference/inference.py,sha256=FUTlgyy676IsFhNH6PTRdnGz6aOXXH_5RtRKqxX8Pss,169801
 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
@@ -980,7 +980,7 @@ supervisely/nn/tracker/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 supervisely/nn/tracker/utils/gmc.py,sha256=3JX8979H3NA-YHNaRQyj9Z-xb9qtyMittPEjGw8y2Jo,11557
 supervisely/nn/tracker/utils/kalman_filter.py,sha256=eSFmCjM0mikHCAFvj-KCVzw-0Jxpoc3Cfc2NWEjJC1Q,17268
 supervisely/nn/training/__init__.py,sha256=gY4PCykJ-42MWKsqb9kl-skemKa8yB6t_fb5kzqR66U,111
-supervisely/nn/training/train_app.py,sha256=mUEXGE9rQZKyVCrX1qjyDFOEuKqdVVa5jQ0EnP0gBGI,106273
+supervisely/nn/training/train_app.py,sha256=GsE8uh_PnExFceEyrmy0ubccbgSf96KCToyP8BXodI0,106666
 supervisely/nn/training/gui/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 supervisely/nn/training/gui/classes_selector.py,sha256=Bpp-RFDQqcZ0kLJmS6ZnExkdscWwRusvF4vbWjEsKlQ,3926
 supervisely/nn/training/gui/gui.py,sha256=QDeWa6iow3z3EwLZcHlQ48d4Ur6hCmedNvCod-8qZ-k,27978
@@ -1082,9 +1082,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.339.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.339.dist-info/METADATA,sha256=jUY-G7GqpZ9Qe4qSdlGrj8lMGoV8P386z114kCWh7gk,33596
-supervisely-6.73.339.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-supervisely-6.73.339.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.339.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.339.dist-info/RECORD,,
+supervisely-6.73.341.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.341.dist-info/METADATA,sha256=1CoprezNuu8F9RHjpci14MSHrSqIsxBLfOZolC6GNyM,33596
+supervisely-6.73.341.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+supervisely-6.73.341.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.341.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.341.dist-info/RECORD,,