fix :169 miss 's'

zhiqwang · Jan 10, 2022 · 585d302 · 585d302
1 parent c9b62db
commit 585d302
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/export_model.py b/tools/export_model.py
@@ -164,7 +164,7 @@ def cli_main():
         model = YOLOv5.load_from_yolov5(
             checkpoint_path,
             size=tuple(image_size),
-            core_thresh=args.score_thresh,
+            score_thresh=args.score_thresh,
             version=args.version,
         )
         model.eval()