Prevent using too low confidence thresholds in detection (#4018)

sovrasov · web-flow · commit 026b5f081b98 · 2024-10-14T14:45:12.000+09:00
Prevent writing too low confidence thresholds to MAPI configuration
diff --git a/src/otx/core/model/detection.py b/src/otx/core/model/detection.py
@@ -280,7 +280,7 @@ def _export_parameters(self) -> TaskLevelExportParameters:
         return super()._export_parameters.wrap(
             model_type="ssd",
             task_type="detection",
-            confidence_threshold=self.hparams.get("best_confidence_threshold", None),
+            confidence_threshold=max(0.35, self.hparams.get("best_confidence_threshold", 0.35)),
             iou_threshold=0.5,
             tile_config=self.tile_config if self.tile_config.enable_tiler else None,
         )

Original file line number	Diff line number	Diff line change
`@@ -280,7 +280,7 @@ def _export_parameters(self) -> TaskLevelExportParameters:`
`280`	`280`	`return super()._export_parameters.wrap(`
`281`	`281`	`model_type="ssd",`
`282`	`282`	`task_type="detection",`
`283`		`- confidence_threshold=self.hparams.get("best_confidence_threshold", None),`
	`283`	`+ confidence_threshold=max(0.35, self.hparams.get("best_confidence_threshold", 0.35)),`
`284`	`284`	`iou_threshold=0.5,`
`285`	`285`	`tile_config=self.tile_config if self.tile_config.enable_tiler else None,`
`286`	`286`	`)`