diff --git a/fast_rcnn_test.py b/fast_rcnn_test.py index 55c35c68f95297f6f5ec76bc50c012499fca125b..cecf6fc093151aad18aaa1d45d2ee813a5ce9c2d 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()