added progress bar

Signed-off-by: Mpho Mphego <mpho112@gmail.com>

Author: mmphego

main.py

@@ -137,7 +137,9 @@ def main(args):
     video_feed = InputFeeder(input_file=args.input)
 
     for frame in video_feed.next_frame():
-        video_feed.show(frame)
+        if args.debug:
+            video_feed.show(frame)
+
     video_feed.close()
 
 

src/input_feeder.py

@@ -2,6 +2,7 @@
 import os
 
 import cv2
+from tqdm import tqdm
 
 from loguru import logger
 from numpy import ndarray
@@ -38,6 +39,7 @@ def __init__(self, input_file=None):
             assert isinstance(self._input_type, str)
         except AssertionError:
             self._input_type = ""
+        self._progress_bar = None
         self.load_data()
 
     def load_data(self):
@@ -70,6 +72,12 @@ def video_len(self):
     def fps(self):
         return int(self.cap.get(cv2.CAP_PROP_FPS))
 
+    @property
+    def progress_bar(self):
+        if not self._progress_bar:
+            self._progress_bar = tqdm(total=int(self.video_len - self.fps + 1))
+        return self._progress_bar
+
     def show(self, frame, frame_name="video"):
         cv2.imshow(frame_name, frame)
 
@@ -86,6 +94,7 @@ def write_video(self, output_path=".", filename="output_video.mp4"):
     def next_frame(self, quit_key="q"):
         """Returns the next image from either a video file or webcam."""
         while self.cap.isOpened():
+            self.progress_bar.update(1)
             flag = False
             for _ in range(1):
                 flag, frame = self.cap.read()
@@ -104,5 +113,7 @@ def close(self):
         """Closes the VideoCapture."""
         if "image" in self._input_type:
             self.cap.release()
+        if self.progress_bar:
+            self.progress_bar.close()
         cv2.destroyAllWindows()
         logger.info("============ CleanUp! ============")