Skip to content

Commit fe464af

Browse files
committed
Update test_cfg
1 parent f984400 commit fe464af

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/otx/algo/detection/rtmdet.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -220,10 +220,10 @@ def _build_model(self, num_classes: int) -> RTMDet:
220220

221221
test_cfg = DictConfig(
222222
{
223-
"nms": {"type": "nms", "iou_threshold": 0.65},
224-
"score_thr": 0.001,
223+
"nms": {"type": "nms", "iou_threshold": 0.5},
224+
"score_thr": 0.05,
225225
"mask_thr_binary": 0.5,
226-
"max_per_img": 300,
226+
"max_per_img": 100,
227227
"min_bbox_size": 0,
228228
"nms_pre": 30000,
229229
},

0 commit comments

Comments
 (0)