Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-09-11 13:15:16 +02:00
parent a31b1489a4
commit 919aff828e
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
train.py
View File

@ -35,6 +35,11 @@ hyp = {'giou': 1.582, # giou loss gain
'scale': 0.1059, # image scale (+/- gain) 'scale': 0.1059, # image scale (+/- gain)
'shear': 0.5768} # image shear (+/- deg) 'shear': 0.5768} # image shear (+/- deg)
if os.path.exists('hyp.txt'): # overwrite hyp if hyp.txt is found
x = np.loadtxt('hyp.txt')
for i, k in enumerate(hyp.keys()):
hyp[k] = x[i]
def train(): def train():
cfg = opt.cfg cfg = opt.cfg