Merge pull request #59 from gshubham533/master

tqdm progress bar bug Thanks a lot!
JustinShenk · Jul 26, 2023 · eea2c8e · eea2c8e
2 parents 8dc4584 + 25e06ae
commit eea2c8e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/fer/classes.py b/src/fer/classes.py
@@ -294,8 +294,12 @@ def analyze(
         if save_video:
             self.videowriter = self._save_video(outfile, fps, width, height)
 
+        total_frames = length
+        if frequency > 1:
+            total_frames = length/frequency
+
         with logging_redirect_tqdm():
-            pbar = tqdm(total=length, unit="frames")
+            pbar = tqdm(total=total_frames, unit="frames")
 
         while self.cap.isOpened():
             ret, frame = self.cap.read()