Add progress bar in image and video demo.

demo/image_demo.py

@@ -5,6 +5,7 @@
 
 import mmcv
 from mmdet.apis import inference_detector, init_detector
+from mmengine.utils import ProgressBar
 
 from mmyolo.registry import VISUALIZERS
 from mmyolo.utils import register_all_modules
@@ -85,6 +86,8 @@ def main(args):
     visualizer.dataset_meta = model.dataset_meta
 
     files = LoadFiles(args.img)
+    progress_bar = ProgressBar(len(files))
+
     for file in files:
         result = inference_detector(model, file)
         img = mmcv.imread(file)
@@ -99,6 +102,7 @@ def main(args):
             wait_time=0,
             out_file=os.path.join(args.out_path, file_name),
             pred_score_thr=args.score_thr)
+        progress_bar.update()
 
 
 if __name__ == '__main__':

demo/video_demo.py

@@ -4,6 +4,7 @@
 import cv2
 import mmcv
 from mmdet.apis import inference_detector, init_detector
+from mmengine.utils import ProgressBar
 
 from mmyolo.registry import VISUALIZERS
 from mmyolo.utils import register_all_modules
@@ -51,6 +52,8 @@ def main():
             args.out, fourcc, video_reader.fps,
             (video_reader.width, video_reader.height))
 
+    progress_bar = ProgressBar(len(video_reader))
+
     for frame in video_reader:
         result = inference_detector(model, frame)
         if args.show:
@@ -68,6 +71,8 @@ def main():
                 pred_score_thr=args.score_thr)
         if args.out:
             video_writer.write(visualizer.get_image())
+        progress_bar.update()
+
     if video_writer:
         video_writer.release()
     cv2.destroyAllWindows()