diff --git a/utils/general.py b/utils/general.py index fa9dd5e67366..b5c4af386b91 100755 --- a/utils/general.py +++ b/utils/general.py @@ -337,7 +337,7 @@ def compute_ap(recall, precision): # Integrate area under curve method = 'interp' # methods: 'continuous', 'interp' if method == 'interp': - x = np.linspace(0, 1, 1001) # 101-point interp (COCO) + x = np.linspace(0, 1, 101) # 101-point interp (COCO) ap = np.trapz(np.interp(x, mrec, mpre), x) # integrate else: # 'continuous' i = np.where(mrec[1:] != mrec[:-1])[0] # points where x axis (recall) changes