Commit caa4a46f authored by Ross Girshick's avatar Ross Girshick

name fix

parent b0a9ffe1
......@@ -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)
......@@ -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)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment