refactor prepare_segmentation_data

supervisely-ecosystem · Dec 13, 2024 · 1435e4a · 1435e4a
1 parent a40a544
commit 1435e4a
Show file tree

Hide file tree

Showing 3 changed files with 223 additions and 166 deletions.
diff --git a/train/src/sly_functions.py b/train/src/sly_functions.py
@@ -13,5 +13,5 @@ def get_eval_results_dir_name(api, task_id, project_info):
     task_info = api.task.get_info_by_id(task_id)
     task_dir = f"{task_id}_{task_info['meta']['app']['name']}"
     eval_res_dir = f"/model-benchmark/{project_info.id}_{project_info.name}/{task_dir}/"
-    eval_res_dir = api.storage.get_free_dir_name(sly.env.team_id(), eval_res_dir)
+    eval_res_dir = api.file.get_free_dir_name(sly.env.team_id(), eval_res_dir)
     return eval_res_dir
diff --git a/train/src/ui/monitoring.html b/train/src/ui/monitoring.html
@@ -125,6 +125,13 @@
 		</div>
 		<el-progress :percentage="data.progressPercentBenchmark"></el-progress>
 	</div>
+    <div v-if="data.progressTqdm" class="mt10">
+		<div style="color: #20a0ff">
+			{{data.progressTqdm}}: {{data.progressCurrentTqdm}} /
+			{{data.progressTotalTqdm}}
+		</div>
+		<el-progress :percentage="data.progressPercentTqdm"></el-progress>
+	</div>
 	<div v-if="data.progressEpoch" class="mt10">
 		<div style="color: #20a0ff">
 			{{data.progressEpoch}}: {{data.progressCurrentEpoch}} /