mAP recorded during training

This commit is contained in:
Glenn Jocher
2018-11-14 15:14:41 +00:00
parent 9dbc3ec1c4
commit 45c5567723
4 changed files with 107 additions and 96 deletions
+1 -1
View File
@@ -11,7 +11,7 @@ gsutil cp gs://ultralytics/yolov3.pt yolov3/weights
python3 detect.py
# Test
python3 test.py -img_size 416 -weights_path weights/latest.pt -conf_thres 0.1
python3 test.py -img_size 416 -weights_path weights/latest.pt -conf_thres 0.5
# Download and Test
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3
+1 -1
View File
@@ -435,7 +435,7 @@ def plot_results():
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']
s = ['X', 'Y', 'Width', 'Height', 'Objectness', 'Classification', 'Total Loss', 'Precision', 'Recall', 'mAP']
for f in ('results.txt',):
results = np.loadtxt(f, usecols=[2, 3, 4, 5, 6, 7, 8, 9, 10]).T
for i in range(9):