From a2279a65856fbb98cbb326c46eb7297b6e7b5095 Mon Sep 17 00:00:00 2001 From: Harim Kang Date: Thu, 16 May 2024 09:24:17 +0900 Subject: [PATCH] Update instance_segmentation.py --- src/otx/core/model/instance_segmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/otx/core/model/instance_segmentation.py b/src/otx/core/model/instance_segmentation.py index 8cea389d233..9720b4da6bf 100644 --- a/src/otx/core/model/instance_segmentation.py +++ b/src/otx/core/model/instance_segmentation.py @@ -113,7 +113,7 @@ def _export_parameters(self) -> TaskLevelExportParameters: return super()._export_parameters.wrap( model_type="MaskRCNN", task_type="instance_segmentation", - confidence_threshold=self.hparams.get("best_confidence_threshold", None), + confidence_threshold=self.hparams.get("best_confidence_threshold", 0.05), iou_threshold=0.5, tile_config=self.tile_config if self.tile_config.enable_tiler else None, )