Procházet zdrojové kódy
Ensures weights/ dir exists
Allows train.py to be run outside of yolov5/ directory.5.0
Lauritzen Kasper Primdal
před 4 roky
1 změnil soubory, kde provedl 1 přidání a 0 odebrání
+ 1
- 0
train.py
Zobrazit soubor
| @@ -20,6 +20,7 @@ except: | |||
| mixed_precision = False # not installed | |||
| wdir = 'weights' + os.sep # weights dir | |||
| os.makedirs(wdir, exist_ok=True) | |||
| last = wdir + 'last.pt' | |||
| best = wdir + 'best.pt' | |||
| results_file = 'results.txt' | |||
Načítá se…