This commit is contained in:
Glenn Jocher
2019-02-18 15:51:35 +01:00
parent 6deda82384
commit c535a8699a
2 changed files with 24 additions and 17 deletions
+1 -1
View File
@@ -443,7 +443,7 @@ def plot_results():
plt.figure(figsize=(16, 8))
s = ['X', 'Y', 'Width', 'Height', 'Confidence', 'Classification', 'Total Loss', 'mAP', 'Recall', 'Precision']
files = sorted(glob.glob('results.txt'))
files = sorted(glob.glob('results*.txt'))
for f in files:
results = np.loadtxt(f, usecols=[2, 3, 4, 5, 6, 7, 8, 11, 12, 13]).T # column 13 is mAP
n = results.shape[1]