From 9e97c4cadb3def7222c2e7462864afe2e6dc3b6e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 29 Jan 2020 11:58:32 -0800 Subject: [PATCH] updates --- utils/utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index c83ebb07..2b3ff82b 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -216,9 +216,11 @@ def ap_per_class(tp, conf, pred_cls, target_cls): # Plot # fig, ax = plt.subplots(1, 1, figsize=(4, 4)) - # ax.plot(np.concatenate(([0.], recall)), np.concatenate(([0.], precision))) - # ax.set_title('YOLOv3-SPP'); ax.set_xlabel('Recall'); ax.set_ylabel('Precision') - # ax.set_xlim(0, 1) + # ax.plot(recall, precision) + # ax.set_xlabel('Recall') + # ax.set_ylabel('Precision') + # ax.set_xlim(0, 1.01) + # ax.set_ylim(0, 1.01) # fig.tight_layout() # fig.savefig('PR_curve.png', dpi=300)