[Fix] fix bug in inferencer demo (#2966)

open-mmlab · Mar 26, 2024 · 7e3340a · 7e3340a
1 parent e0eb5d4
commit 7e3340a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/inferencer_demo.py b/demo/inferencer_demo.py
@@ -83,7 +83,7 @@ def parse_args():
     # selected pose2d model
     args, _ = parser.parse_known_args()
     for model in POSE2D_SPECIFIC_ARGS:
-        if model in args.pose2d:
+        if args.pose2d is not None and model in args.pose2d:
             filter_args.update(POSE2D_SPECIFIC_ARGS[model])
             break