From 585d302c4f6a738b4a66d3380accd0a2477687a7 Mon Sep 17 00:00:00 2001 From: lzmisscc <1028741731@qq.com> Date: Mon, 10 Jan 2022 18:02:37 +0800 Subject: [PATCH] fix :169 miss 's' --- tools/export_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/export_model.py b/tools/export_model.py index 5f0f40c7..ee1ef198 100644 --- 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()