Allows train.py to be run outside of yolov5/ directory.5.0
mixed_precision = False # not installed | mixed_precision = False # not installed | ||||
wdir = 'weights' + os.sep # weights dir | wdir = 'weights' + os.sep # weights dir | ||||
os.makedirs(wdir, exist_ok=True) | |||||
last = wdir + 'last.pt' | last = wdir + 'last.pt' | ||||
best = wdir + 'best.pt' | best = wdir + 'best.pt' | ||||
results_file = 'results.txt' | results_file = 'results.txt' |