From ccb5051f7f485a2db1affeecd2297f49a1bb06dd Mon Sep 17 00:00:00 2001 From: Laughing-q <1185102784@qq.com> Date: Thu, 31 Oct 2024 17:51:13 +0800 Subject: [PATCH] Update exporter.py --- ultralytics/engine/exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 874d60cc3..0d0269e22 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -1189,7 +1189,7 @@ class Exporter: quant_model = NMSWrapper( model=quant_model, - score_threshold=self.args.conf, + score_threshold=self.args.conf or 0.001, iou_threshold=self.args.iou, max_detections=self.args.max_det, ).to(self.device)