From 0e8d27f2468b32fefc08610562f5714370895efc Mon Sep 17 00:00:00 2001 From: Dahun Kim Date: Tue, 31 Aug 2021 18:37:29 +0900 Subject: [PATCH] update config --- configs/oln_box/oln_box.py | 5 ++--- mmdet/models/dense_heads/oln_rpn_head.py | 3 +++ 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/configs/oln_box/oln_box.py b/configs/oln_box/oln_box.py index c5db0c0..23fba31 100644 --- a/configs/oln_box/oln_box.py +++ b/configs/oln_box/oln_box.py @@ -105,8 +105,7 @@ model = dict( nms_pre=2000, nms_post=2000, max_num=2000, - # rpn_nms is not used. - nms_thr=1.0, + nms_thr=1.0, # nms is not used. min_bbox_size=0), rcnn=dict( assigner=dict( @@ -130,7 +129,7 @@ model = dict( nms_pre=2000, nms_post=2000, max_num=2000, - nms_thr=0.7, + nms_thr=1.0, # nms is not used. min_bbox_size=0), rcnn=dict( score_thr=0.0, diff --git a/mmdet/models/dense_heads/oln_rpn_head.py b/mmdet/models/dense_heads/oln_rpn_head.py index fc09ce5..3bd53e6 100644 --- a/mmdet/models/dense_heads/oln_rpn_head.py +++ b/mmdet/models/dense_heads/oln_rpn_head.py @@ -688,6 +688,9 @@ class OlnRPNHead(RPNHead): ids = ids[valid_inds] nms_cfg = dict(type='nms', iou_threshold=cfg.nms_thr) + + import pdb; pdb.set_trace() + if nms_cfg in [0.0, 1.0]: # No NMS: dets = torch.cat([proposals, scores.unsqueeze(1)], 1)