From d351717264fa597969642ca98b4950ab7904164d Mon Sep 17 00:00:00 2001
From: Ross Girshick <ross.girshick@gmail.com>
Date: Tue, 24 Mar 2015 14:51:17 -0700
Subject: [PATCH] fix nms call
---
fast_rcnn_test.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/fast_rcnn_test.py b/fast_rcnn_test.py
index 55c35c6..cecf6fc 100644
--- a/fast_rcnn_test.py
+++ b/fast_rcnn_test.py
@@ -251,7 +251,7 @@ def test_net(net, imdb):
.astype(np.float32, copy=False)
if 0:
- keep = utils.nms.nms(all_boxes[j][i], 0.3)
+ keep = utils.cython_nms.nms(all_boxes[j][i], 0.3)
_vis_detections(im, imdb.classes[j], all_boxes[j][i][keep, :])
_t['misc'].toc()
--
GitLab