apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Reinstate PR curve sentinel values (#1598)

Browse Source
This commit is contained in:
Glenn Jocher 2020-12-08 17:40:19 -08:00 committed by GitHub
parent ce9feb42b4
commit 7e846c7d3c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
utils/metrics.py
View File

@ -77,18 +77,17 @@ def ap_per_class(tp, conf, pred_cls, target_cls, plot=False, save_dir='precision
def compute_ap(recall, precision):
""" Compute the average precision, given the recall and precision curves.
Source: https://github.com/rbgirshick/py-faster-rcnn.
""" Compute the average precision, given the recall and precision curves
# Arguments
recall: The recall curve (list).
precision: The precision curve (list).
recall: The recall curve (list)
precision: The precision curve (list)
# Returns
The average precision as computed in py-faster-rcnn.
Average precision, precision curve, recall curve
"""
# Append sentinel values to beginning and end
mrec = recall # np.concatenate(([0.], recall, [recall[-1] + 1E-3]))
mpre = precision # np.concatenate(([0.], precision, [0.]))
mrec = np.concatenate(([0.], recall, [recall[-1] + 0.01]))
mpre = np.concatenate(([1.], precision, [0.]))
# Compute the precision envelope
mpre = np.flip(np.maximum.accumulate(np.flip(mpre)))