QoL fix: Progress bar should now respect batch length.

AnonymousCervine · Feb 3, 2023 · 5d49f71 · 5d49f71
1 parent 390a065
commit 5d49f71
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/scripts/depth-image-io.py b/scripts/depth-image-io.py
@@ -199,6 +199,12 @@ def run(self, p, input_depth_img, show_depth, batch_img_input, batch_depth_input
 
             processed_output = None
 
+            if batch_many_to_many:
+                num_outputs = len(batch_color) * len(batch_depth)
+            else:
+                num_outputs = max(len(batch_color), len(batch_depth))
+            shared.state.job_count = num_outputs * p.n_iter
+
             def batch_item(depth_img, color_img):
                 if color_img:
                     # override input images