Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2020-01-29 11:58:32 -08:00
parent f405d45043
commit 9e97c4cadb
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
utils/utils.py
View File

@ -216,9 +216,11 @@ def ap_per_class(tp, conf, pred_cls, target_cls):
# Plot # Plot
# fig, ax = plt.subplots(1, 1, figsize=(4, 4)) # fig, ax = plt.subplots(1, 1, figsize=(4, 4))
# ax.plot(np.concatenate(([0.], recall)), np.concatenate(([0.], precision))) # ax.plot(recall, precision)
# ax.set_title('YOLOv3-SPP'); ax.set_xlabel('Recall'); ax.set_ylabel('Precision') # ax.set_xlabel('Recall')
# ax.set_xlim(0, 1) # ax.set_ylabel('Precision')
# ax.set_xlim(0, 1.01)
# ax.set_ylim(0, 1.01)
# fig.tight_layout() # fig.tight_layout()
# fig.savefig('PR_curve.png', dpi=300) # fig.savefig('PR_curve.png', dpi=300)