From bb15adc0d1f4003c6b6def78737f831105efe8e9 Mon Sep 17 00:00:00 2001 From: Vladislav Sovrasov Date: Wed, 6 Nov 2024 01:07:39 +0100 Subject: [PATCH] Revert the old workaround for detection confidence threshold (#4096) Revert the old workaround --- src/otx/core/model/detection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/otx/core/model/detection.py b/src/otx/core/model/detection.py index 1b3dad78618..064811189ce 100644 --- 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=max(0.35, self.hparams.get("best_confidence_threshold", 0.35)), + confidence_threshold=self.hparams.get("best_confidence_threshold", None), iou_threshold=0.5, tile_config=self.tile_config if self.tile_config.enable_tiler else None, )