Adding total frames to tqdm

DiogenesAnalytics · Feb 3, 2024 · 996d244 · 996d244
1 parent b7d69f8
commit 996d244
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/frame_sampling/strategy.py b/src/frame_sampling/strategy.py
@@ -66,6 +66,7 @@ def _create_stream(
             enumerate(
                 tqdm(
                     container.decode(stream),
+                    total=stream.frames,
                     desc=f"Sampling {video_name}",
                     leave=False,
                     delay=self.iter_frames_progress_delay,