Update exporter.py

1 month ago · ccb5051f7f
parent ef8c59cf50
commit ccb5051f7f
1 changed files with 1 additions and 1 deletions
--- 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)