From f2e46b22ad41f60c2cd5b205b4237f6baac965fb Mon Sep 17 00:00:00 2001 From: Kai KANG Date: Mon, 9 May 2016 15:09:49 +0800 Subject: [PATCH] Fix a return-value bug in eval_recall.py --- tools/eval_recall.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/tools/eval_recall.py b/tools/eval_recall.py index b1a59dc27..3254d50fa 100755 --- a/tools/eval_recall.py +++ b/tools/eval_recall.py @@ -46,15 +46,14 @@ def parse_args(): raw_data = sio.loadmat(filename)['aboxes'].ravel() candidate_boxes = raw_data - ar, gt_overlaps, recalls, thresholds = \ - imdb.evaluate_recall(candidate_boxes=candidate_boxes) + res = imdb.evaluate_recall(candidate_boxes=candidate_boxes) print 'Method: {}'.format(args.method) - print 'AverageRec: {:.3f}'.format(ar) + print 'AverageRec: {:.3f}'.format(res['ar']) def recall_at(t): - ind = np.where(thresholds > t - 1e-5)[0][0] - assert np.isclose(thresholds[ind], t) - return recalls[ind] + ind = np.where(res['thresholds'] > t - 1e-5)[0][0] + assert np.isclose(res['thresholds'][ind], t) + return res['recalls'][ind] print 'Recall@0.5: {:.3f}'.format(recall_at(0.5)) print 'Recall@0.6: {:.3f}'.format(recall_at(0.6)) @@ -62,7 +61,7 @@ def recall_at(t): print 'Recall@0.8: {:.3f}'.format(recall_at(0.8)) print 'Recall@0.9: {:.3f}'.format(recall_at(0.9)) # print again for easy spreadsheet copying - print '{:.3f}'.format(ar) + print '{:.3f}'.format(res['ar']) print '{:.3f}'.format(recall_at(0.5)) print '{:.3f}'.format(recall_at(0.6)) print '{:.3f}'.format(recall_at(0.7))