diff --git a/plot_PR_curve.py b/plot_PR_curve.py index 5d19b82b8bf9d50835ff3c09023324ec8c565a07..4b95f28a8a548160abb1c5ad8f1dd821059f9b4c 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()