minor fix

pull/3/head
Dahun Kim 2021-08-31 20:48:51 +09:00
parent 644cdd4e32
commit 7636de2df1
2 changed files with 1 additions and 4 deletions

View File

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

View File

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