From 7121bef9f44430659033abb8a2bfb161b802dc5b Mon Sep 17 00:00:00 2001 From: pfjaeger Date: Wed, 24 Oct 2018 11:40:58 +0200 Subject: [PATCH] NMS bugfix --- nms/pth_nms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nms/pth_nms.py b/nms/pth_nms.py index fcaea7b..8863ffa 100644 --- a/nms/pth_nms.py +++ b/nms/pth_nms.py @@ -40,7 +40,7 @@ def pth_nms(dets, thresh): order = scores.sort(0, descending=True)[1] # order = torch.from_numpy(np.ascontiguousarray(scores.cpu().numpy().argsort()[::-1])).long().cuda() - dets = dets[order].contiguous() + dets_temp = dets_temp[order].contiguous() keep = torch.LongTensor(dets.size(0)) num_out = torch.LongTensor(1)