From 4ec6648f14cbcc2a454903f8aa6ae0b892dfd3fd Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephane.chavin@lis-lab.fr>
Date: Mon, 2 Dec 2024 16:36:29 +0100
Subject: [PATCH] save the metrics (precision / recall)
---
yolov5/utils/metrics.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/yolov5/utils/metrics.py b/yolov5/utils/metrics.py
index 95f364c..25cc678 100755
--- a/yolov5/utils/metrics.py
+++ b/yolov5/utils/metrics.py
@@ -333,6 +333,8 @@ def plot_pr_curve(px, py, ap, save_dir=Path('pr_curve.png'), names=()):
ax.set_ylim(0, 1)
ax.legend(bbox_to_anchor=(1.04, 1), loc='upper left')
ax.set_title('Precision-Recall Curve')
+ metrics_dir = save_dir.parent / 'class_metrics.npy'
+ np.save(metrics_dir, py)
fig.savefig(save_dir, dpi=250)
plt.close(fig)
--
GitLab