Skip to content

Commit

Permalink
Revert the old workaround for detection confidence threshold (#4096)
Browse files Browse the repository at this point in the history
Revert the old workaround
  • Loading branch information
sovrasov authored Nov 6, 2024
1 parent 15746ea commit bb15adc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/otx/core/model/detection.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
)
Expand Down

0 comments on commit bb15adc

Please sign in to comment.