[Bug Fix] video_demo bug fixed for issue#11353 (#11451)

open-mmlab · Feb 1, 2024 · 76e23d2 · 76e23d2
1 parent 44ebd17
commit 76e23d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/video_demo.py b/demo/video_demo.py
@@ -58,7 +58,7 @@ def main():
             args.out, fourcc, video_reader.fps,
             (video_reader.width, video_reader.height))
 
-    for frame in track_iter_progress(video_reader):
+    for frame in track_iter_progress((video_reader, len(video_reader))):
         result = inference_detector(model, frame, test_pipeline=test_pipeline)
         visualizer.add_datasample(
             name='video',