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

updates

This commit is contained in:
Glenn Jocher 2019-04-18 22:31:05 +02:00
parent 9b6347ac6c
commit cc50757d95
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
utils/utils.py
View File

@ -520,7 +520,8 @@ def plot_results(start=0, stop=0): # from utils.utils import *; plot_results()
# Plot training results files 'results*.txt' # Plot training results files 'results*.txt'
# import os; os.system('wget https://storage.googleapis.com/ultralytics/yolov3/results_v3.txt') # import os; os.system('wget https://storage.googleapis.com/ultralytics/yolov3/results_v3.txt')
fig = plt.figure(figsize=(14, 7)) fig, ax = plt.subplots(2, 5, figsize=(14, 7))
ax = ax.ravel()
s = ['X + Y', 'Width + Height', 'Confidence', 'Classification', 'Train Loss', 'Precision', 'Recall', 'mAP', 'F1', s = ['X + Y', 'Width + Height', 'Confidence', 'Classification', 'Train Loss', 'Precision', 'Recall', 'mAP', 'F1',
'Test Loss'] 'Test Loss']
for f in sorted(glob.glob('results*.txt') + glob.glob('../../Downloads/results*.txt')): for f in sorted(glob.glob('results*.txt') + glob.glob('../../Downloads/results*.txt')):
@ -528,10 +529,8 @@ def plot_results(start=0, stop=0): # from utils.utils import *; plot_results()
n = results.shape[1] # number of rows n = results.shape[1] # number of rows
x = range(start, min(stop, n) if stop else n) x = range(start, min(stop, n) if stop else n)
for i in range(10): for i in range(10):
plt.subplot(2, 5, i + 1) ax[i].plot(x, results[i, x], marker='.', label=f.replace('.txt', ''))
plt.plot(x, results[i, x], marker='.', label=f.replace('.txt', '')) ax[i].set_title(s[i])
plt.title(s[i]) ax[0].legend()
if i == 0:
plt.legend()
fig.tight_layout() fig.tight_layout()
fig.savefig('results.png', dpi=300) fig.savefig('results.png', dpi=300)