Просмотр исходного кода

Ensures weights/ dir exists

Allows train.py to be run outside of yolov5/ directory.
5.0
Lauritzen Kasper Primdal 4 лет назад
Родитель
Сommit
c3d4d321d3
1 измененных файлов: 1 добавлений и 0 удалений
  1. +1
    -0
      train.py

+ 1
- 0
train.py Просмотреть файл

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

Загрузка…
Отмена
Сохранить