From 431bbfde8edbe510b58b774265710a1a1f84e67e Mon Sep 17 00:00:00 2001
From: Almaz <79905215+almazgimaev@users.noreply.github.com>
Date: Thu, 26 Sep 2024 23:17:46 +0200
Subject: [PATCH] Relative img paths (#64)

---
 train/src/main.py          | 18 +++++++++---------
 train/src/sly_to_yolov8.py |  2 +-
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/train/src/main.py b/train/src/main.py
index a51d61d..2c5ba9e 100644
--- 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,20 +1828,20 @@ 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")
@@ -1849,7 +1849,7 @@ def train_model():
             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")
@@ -1857,7 +1857,7 @@ def train_model():
             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")
@@ -1865,7 +1865,7 @@ def train_model():
             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
index 34cfc65..f5ad0c3 100644
--- 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