Relative img paths (#64)

supervisely-ecosystem · Sep 26, 2024 · 431bbfd · 431bbfd
1 parent 7f9154b
commit 431bbfd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/train/src/main.py b/train/src/main.py
@@ -1681,7 +1681,7 @@ def on_results_file_changed(filepath, pbar):
             tf_train_batches_info = api.file.upload(
                 team_id, local_train_batches_path, remote_train_batches_path
             )
-            train_batches_gallery.append(tf_train_batches_info.full_storage_url)
+            train_batches_gallery.append(tf_train_batches_info.storage_path)
             if x == 1:
                 train_batches_gallery_f.show()
 
@@ -1795,7 +1795,7 @@ def train_model():
             )
             tf_labels_info = api.file.upload(team_id, labels_path, remote_labels_path)
             val_batch_labels_id = val_batches_gallery.append(
-                image_url=tf_labels_info.full_storage_url,
+                image_url=tf_labels_info.storage_path,
                 title="labels",
             )
         preds_path = os.path.join(local_artifacts_dir, f"val_batch{i}_pred.jpg")
@@ -1805,7 +1805,7 @@ def train_model():
             )
             tf_preds_info = api.file.upload(team_id, preds_path, remote_preds_path)
             val_batch_preds_id = val_batches_gallery.append(
-                image_url=tf_preds_info.full_storage_url,
+                image_url=tf_preds_info.storage_path,
                 title="predictions",
             )
         if val_batch_labels_id and val_batch_preds_id:
@@ -1828,44 +1828,44 @@ def train_model():
             team_id, confusion_matrix_path, remote_confusion_matrix_path
         )
         if not app.is_stopped():
-            additional_gallery.append(tf_confusion_matrix_info.full_storage_url)
+            additional_gallery.append(tf_confusion_matrix_info.storage_path)
             additional_gallery_f.show()
     pr_curve_path = os.path.join(local_artifacts_dir, "PR_curve.png")
     if os.path.exists(pr_curve_path):
         remote_pr_curve_path = os.path.join(remote_images_path, "PR_curve.png")
         tf_pr_curve_info = api.file.upload(team_id, pr_curve_path, remote_pr_curve_path)
         if not app.is_stopped():
-            additional_gallery.append(tf_pr_curve_info.full_storage_url)
+            additional_gallery.append(tf_pr_curve_info.storage_path)
     f1_curve_path = os.path.join(local_artifacts_dir, "F1_curve.png")
     if os.path.exists(f1_curve_path):
         remote_f1_curve_path = os.path.join(remote_images_path, "F1_curve.png")
         tf_f1_curve_info = api.file.upload(team_id, f1_curve_path, remote_f1_curve_path)
         if not app.is_stopped():
-            additional_gallery.append(tf_f1_curve_info.full_storage_url)
+            additional_gallery.append(tf_f1_curve_info.storage_path)
     box_f1_curve_path = os.path.join(local_artifacts_dir, "BoxF1_curve.png")
     if os.path.exists(box_f1_curve_path):
         remote_box_f1_curve_path = os.path.join(remote_images_path, "BoxF1_curve.png")
         tf_box_f1_curve_info = api.file.upload(
             team_id, box_f1_curve_path, remote_box_f1_curve_path
         )
         if not app.is_stopped():
-            additional_gallery.append(tf_box_f1_curve_info.full_storage_url)
+            additional_gallery.append(tf_box_f1_curve_info.storage_path)
     pose_f1_curve_path = os.path.join(local_artifacts_dir, "PoseF1_curve.png")
     if os.path.exists(pose_f1_curve_path):
         remote_pose_f1_curve_path = os.path.join(remote_images_path, "PoseF1_curve.png")
         tf_pose_f1_curve_info = api.file.upload(
             team_id, pose_f1_curve_path, remote_pose_f1_curve_path
         )
         if not app.is_stopped():
-            additional_gallery.append(tf_pose_f1_curve_info.full_storage_url)
+            additional_gallery.append(tf_pose_f1_curve_info.storage_path)
     mask_f1_curve_path = os.path.join(local_artifacts_dir, "MaskF1_curve.png")
     if os.path.exists(mask_f1_curve_path):
         remote_mask_f1_curve_path = os.path.join(remote_images_path, "MaskF1_curve.png")
         tf_mask_f1_curve_info = api.file.upload(
             team_id, mask_f1_curve_path, remote_mask_f1_curve_path
         )
         if not app.is_stopped():
-            additional_gallery.append(tf_mask_f1_curve_info.full_storage_url)
+            additional_gallery.append(tf_mask_f1_curve_info.storage_path)
 
     making_training_vis_f.hide()
     # rename best checkpoint file

diff --git a/train/src/sly_to_yolov8.py b/train/src/sly_to_yolov8.py
@@ -34,7 +34,7 @@ def check_bbox_exist_on_images(api, selected_classes, datasets, project_meta, pr
                                 bbox_exist = True
                                 break
                     if not bbox_exist:
-                        bbox_miss_image_urls.append(img_info.preview_url)
+                        bbox_miss_image_urls.append(img_info.path_original)
                     pbar.update()
     return bbox_miss_image_urls