diff --git a/main.py b/main.py
index cdffc7e..64075e5 100644
--- a/main.py
+++ b/main.py
@@ -97,7 +97,7 @@ def main():
     parser.add_argument("--outdir", required=True, help="图片输出目录")
     parser.add_argument("--device", default="cuda", choices=["cuda", "cpu"], help="推理设备")
     parser.add_argument("--dtype", default="fp16", choices=["fp16", "fp32"], help="推理精度")
-    args, _ = p.parse_known_args()
+    args, _ = parser.parse_known_args()
 
     model_path = args.model
     json_path = Path(args.json)