Ensures weights/ dir exists · c3d4d321d3 - yolov5 - 拓恒技术版本管理系统

ソースを参照

Ensures weights/ dir exists

Allows train.py to be run outside of yolov5/ directory.

5.0

Lauritzen Kasper Primdal 4年前

親

099e6f5ebd

コミット

c3d4d321d3

1個のファイルの変更、1行の追加、0行の削除

分割表示

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