diff --git a/utils/metrics.py b/utils/metrics.py index ddc425910a75..44b9a3c16488 100644 --- a/utils/metrics.py +++ b/utils/metrics.py @@ -91,8 +91,8 @@ def compute_ap(recall, precision): """ # Append sentinel values to beginning and end - mrec = np.concatenate(([0.], recall, [recall[-1] + 0.01])) - mpre = np.concatenate(([1.], precision, [0.])) + mrec = np.concatenate(([0.0], recall, [1.0])) + mpre = np.concatenate(([1.0], precision, [0.0])) # Compute the precision envelope mpre = np.flip(np.maximum.accumulate(np.flip(mpre)))