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

updates

This commit is contained in:
Glenn Jocher 2018-11-30 11:56:38 +01:00
parent 35e445c5da
commit 0240ac44f6
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
utils/utils.py
View File

@ -442,12 +442,28 @@ def plot_results():
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
plt.figure(figsize=(16, 8)) plt.figure(figsize=(16, 8))
s = ['X', 'Y', 'Width', 'Height', 'Objectness', 'Classification', 'Total Loss', 'Precision', 'Recall', 'mAP'] s = ['X', 'Y', 'Width', 'Height', 'Objectness', 'Classification', 'Total Loss', 'Precision', 'Recall', 'mAP']
for f in ('results_d5.txt', 'results_d10.txt', 'results_new.txt', for f in ('results.txt',
): ):
results = np.loadtxt(f, usecols=[2, 3, 4, 5, 6, 7, 8, 9, 10]).T # column 16 is mAP results = np.loadtxt(f, usecols=[2, 3, 4, 5, 6, 7, 8, 17, 18, 16]).T # column 16 is mAP
for i in range(9): for i in range(10):
plt.subplot(2, 5, i + 1) plt.subplot(2, 5, i + 1)
plt.plot(results[i, :250], marker='.', label=f) plt.plot(results[i, :250], marker='.', label=f)
plt.title(s[i]) plt.title(s[i])
if i == 0: if i == 0:
plt.legend() plt.legend()
# def plot_results():
# # Plot YOLO training results file 'results.txt'
# import numpy as np
# import matplotlib.pyplot as plt
# plt.figure(figsize=(16, 8))
# s = ['X', 'Y', 'Width', 'Height', 'Objectness', 'Classification', 'Total Loss', 'Precision', 'Recall', 'mAP']
# for f in ('results_d5.txt', 'results_d10.txt', 'results_64.txt',
# ):
# results = np.loadtxt(f, usecols=[16]).T # column 16 is mAP
# for i in range(1):
# plt.subplot(2, 5, i + 1)
# plt.plot(results, marker='.', label=f)
# plt.title(s[i])
# if i == 0:
# plt.legend()