From ef877f0150b09b9b6c4d19637e8b52f5febe500a Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephane.chavin@lis-lab.fr>
Date: Thu, 3 Apr 2025 10:17:56 +0200
Subject: [PATCH] correct axis
---
plot_PR_curve.py | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/plot_PR_curve.py b/plot_PR_curve.py
index 5d19b82..4b95f28 100644
--- a/plot_PR_curve.py
+++ b/plot_PR_curve.py
@@ -40,7 +40,14 @@ def main(arguments):
label=f'mAP {class_index} = {area_under_curve}',
color=colors[class_index],
linestyle=lines[class_index])
- plt.legend(loc='upper center', bbox_to_anchor=(0.5, -0.05),
+
+ plt.xlabel('Recall')
+ plt.ylabel('Precision')
+
+ plt.xlim([0,1])
+ plt.ylim([0,1])
+
+ plt.legend(loc='upper center', bbox_to_anchor=(0.5, -0.08),
fancybox=True, ncol=8)
plt.tight_layout()
--
GitLab