diff --git a/demo/video_demo.py b/demo/video_demo.py index 6fc36319ae8..f72d617695c 100644 --- 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',