Precision-Recall Curve feature update ultralytics#1206

KMint1819 · Oct 25, 2020 · 1bac5b8 · 1bac5b8
1 parent 04982f9
commit 1bac5b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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