update config
parent
5a18ea0bb0
commit
0e8d27f246
configs/oln_box
mmdet/models/dense_heads
|
@ -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,
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue