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

updates

This commit is contained in:
Glenn Jocher 2019-08-05 15:50:45 +02:00
parent a1a86cd784
commit b2e87d0844
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
utils/utils.py
View File

@ -766,7 +766,7 @@ def plot_results(start=0, stop=0): # from utils.utils import *; plot_results()
fig.savefig('results.png', dpi=200) fig.savefig('results.png', dpi=200)
def plot_results_overlay(start=0, stop=0): # from utils.utils import *; plot_results_overlay() def plot_results_overlay(start=1, stop=0): # from utils.utils import *; plot_results_overlay()
# Plot training results files 'results*.txt', overlaying train and val losses # Plot training results files 'results*.txt', overlaying train and val losses
s = ['train', 'train', 'train', 'Precision', 'mAP', 'val', 'val', 'val', 'Recall', 'F1'] # legends s = ['train', 'train', 'train', 'Precision', 'mAP', 'val', 'val', 'val', 'Recall', 'F1'] # legends
t = ['GIoU', 'Confidence', 'Classification', 'P-R', 'mAP-F1'] # titles t = ['GIoU', 'Confidence', 'Classification', 'P-R', 'mAP-F1'] # titles
@ -781,6 +781,7 @@ def plot_results_overlay(start=0, stop=0): # from utils.utils import *; plot_re
ax[i].plot(x, results[i + 5, x], marker='.', label=s[i + 5]) ax[i].plot(x, results[i + 5, x], marker='.', label=s[i + 5])
ax[i].set_title(t[i]) ax[i].set_title(t[i])
ax[i].legend() ax[i].legend()
ax[i].set_ylabel(f) if i == 0 else None # add filename
fig.tight_layout() fig.tight_layout()
fig.savefig(f.replace('.txt', '.png'), dpi=200) fig.savefig(f.replace('.txt', '.png'), dpi=200)