fix parser

2025-09-03 16:04:18 +08:00
parent eeda961724
commit 96ad5735ac
1 changed files with 1 additions and 1 deletions
--- 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)