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-21 19:22:01 +01:00
parent 7283f52d0c
commit dae9b8f4b5
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
detect.py
View File

@ -145,3 +145,5 @@ def main(opt):
if __name__ == '__main__': if __name__ == '__main__':
torch.cuda.empty_cache() torch.cuda.empty_cache()
main(opt) main(opt)
if platform == 'darwin': # MacOS (local)
os.system('open ' + opt.output_folder)

2
utils/gcp.sh
View File

@ -1,7 +1,7 @@
#!/usr/bin/env bash #!/usr/bin/env bash
# Start # Start
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py -epochs 2
# Resume # Resume
python3 train.py -resume 1 python3 train.py -resume 1

2
utils/utils.py
View File

@ -436,7 +436,7 @@ 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.txt', for f in ('results5.txt','results_new.txt','results3.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, 9, 10]).T # column 16 is mAP
for i in range(9): for i in range(9):