update config

pull/3/head
Dahun Kim 2021-08-31 18:37:29 +09:00
parent 5a18ea0bb0
commit 0e8d27f246
2 changed files with 5 additions and 3 deletions
configs/oln_box
mmdet/models/dense_heads

View File

@ -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,

View File

@ -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)