minor fix
parent
644cdd4e32
commit
7636de2df1
|
@ -129,7 +129,7 @@ model = dict(
|
|||
nms_pre=2000,
|
||||
nms_post=2000,
|
||||
max_num=2000,
|
||||
nms_thr=0.7,
|
||||
nms_thr=0.0, # No nms
|
||||
min_bbox_size=0),
|
||||
rcnn=dict(
|
||||
score_thr=0.0,
|
||||
|
|
|
@ -691,9 +691,6 @@ class OlnRPNHead(RPNHead):
|
|||
|
||||
# No NMS:
|
||||
dets = torch.cat([proposals, scores.unsqueeze(1)], 1)
|
||||
# else:
|
||||
# dets, keep = batched_nms(proposals, scores, ids, nms_cfg)
|
||||
# return dets[:cfg.nms_post]
|
||||
|
||||
return dets
|
||||
|
||||
|
|
Loading…
Reference in New Issue