From caa4a46ff1d5b534d45384c81a4c495156207d82 Mon Sep 17 00:00:00 2001
From: Ross Girshick <ross.girshick@gmail.com>
Date: Wed, 22 Apr 2015 13:33:36 -0700
Subject: [PATCH] name fix
---
src/fast_rcnn_test.py | 4 ++--
tools/extra/reval.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/fast_rcnn_test.py b/src/fast_rcnn_test.py
index 3414acf..1f2f2d1 100644
--- a/src/fast_rcnn_test.py
+++ b/src/fast_rcnn_test.py
@@ -178,7 +178,7 @@ def _vis_detections(im, class_name, dets, thresh=0.3):
plt.title('{} {:.3f}'.format(class_name, score))
plt.pause(1)
-def _apply_nms(all_boxes, thresh):
+def apply_nms(all_boxes, thresh):
num_classes = len(all_boxes)
num_images = len(all_boxes[0])
nms_boxes = [[[] for _ in xrange(num_images)]
@@ -269,7 +269,7 @@ def test_net(net, imdb):
cPickle.dump(all_boxes, f, cPickle.HIGHEST_PROTOCOL)
print 'Applying NMS to all detections'
- nms_dets = _apply_nms(all_boxes, cfg.TEST.NMS)
+ nms_dets = apply_nms(all_boxes, cfg.TEST.NMS)
print 'Evaluating detections'
imdb.evaluate_detections(nms_dets, output_dir)
diff --git a/tools/extra/reval.py b/tools/extra/reval.py
index 2ad0a89..d873534 100755
--- a/tools/extra/reval.py
+++ b/tools/extra/reval.py
@@ -18,7 +18,7 @@ def main(imdb_name, output_dir):
dets = cPickle.load(f)
print 'Applying NMS to all detections'
- nms_dets = fast_rcnn_test._apply_nms(dets, cfg.TEST.NMS)
+ nms_dets = fast_rcnn_test.apply_nms(dets, cfg.TEST.NMS)
print 'Evaluating detections'
imdb.evaluate_detections(nms_dets, output_dir)
--
GitLab