Parcourir la source

Ensures weights/ dir exists

Allows train.py to be run outside of yolov5/ directory.
5.0
Lauritzen Kasper Primdal il y a 4 ans
Parent
révision
c3d4d321d3
1 fichiers modifiés avec 1 ajouts et 0 suppressions
  1. +1
    -0
      train.py

+ 1
- 0
train.py Voir le fichier

@@ -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'

Chargement…
Annuler
Enregistrer