Setting scope for image reader in video_demo (#10868)

open-mmlab · Sep 4, 2023 · af816d3 · af816d3
1 parent a9c4c01
commit af816d3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/demo/video_demo.py b/demo/video_demo.py
@@ -40,7 +40,8 @@ def main():
     model = init_detector(args.config, args.checkpoint, device=args.device)
 
     # build test pipeline
-    model.cfg.test_dataloader.dataset.pipeline[0].type = 'LoadImageFromNDArray'
+    model.cfg.test_dataloader.dataset.pipeline[
+        0].type = 'mmdet.LoadImageFromNDArray'
     test_pipeline = Compose(model.cfg.test_dataloader.dataset.pipeline)
 
     # init visualizer