[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',