minor fix
parent
0e8d27f246
commit
74b18e72f8
|
@ -689,8 +689,6 @@ class OlnRPNHead(RPNHead):
|
|||
|
||||
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