Prevent PR plotting (#1489)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-11-24 00:46:01 +01:00
1 parent 0ddf12e108
commit 7aeef2dca5
1 file changed
+1 -1
+1 -1
View File
@@ -64,7 +64,7 @@ def ap_per_class(tp, conf, pred_cls, target_cls, plot=False, save_dir='precision
# AP from recall-precision curve
for j in range(tp.shape[1]):
ap[ci, j], mpre, mrec = compute_ap(recall[:, j], precision[:, j])
if j == 0:
if plot and (j == 0):
py.append(np.interp(px, mrec, mpre)) # precision at mAP@0.5
# Compute F1 score (harmonic mean of precision and recall)