add total count to progress

supervisely-ecosystem · Dec 2, 2024 · 4a060bc · 4a060bc
1 parent 1a4a56c
commit 4a060bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/convert_sly_to_yolov5.py b/src/convert_sly_to_yolov5.py
@@ -209,7 +209,7 @@ def _add_to_split(image_id, img_name, split_ids, split_image_paths, labels_dir,
 
         ids_to_download = train_ids + val_ids
         paths_to_download = train_image_paths + val_image_paths
-        progress = sly.Progress("Downloading images...")
+        progress = sly.Progress("Downloading images...", total_cnt=len(ids_to_download))
         for batch_ids, batch_paths in zip(
             sly.batched(ids_to_download), sly.batched(paths_to_download)
         ):