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

updates

This commit is contained in:
Glenn Jocher 2019-08-06 14:36:12 +02:00
parent bd2d3cc5d1
commit 0e7cd7e283
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -61,7 +61,7 @@ hyp = {'giou': 1.43, # giou loss gain
'cls': 11.7, # cls loss gain 'cls': 11.7, # cls loss gain
'cls_pw': 4.81, # cls BCELoss positive_weight 'cls_pw': 4.81, # cls BCELoss positive_weight
'obj': 11.5, # obj loss gain 'obj': 11.5, # obj loss gain
'obj_pw': 1.56, # obj BCELoss positive_weight 'obj_pw': 1.0, # obj BCELoss positive_weight
'iou_t': 0.281, # iou training threshold 'iou_t': 0.281, # iou training threshold
'lr0': 0.0013, # initial learning rate 'lr0': 0.0013, # initial learning rate
'lrf': -4., # final LambdaLR learning rate = lr0 * (10 ** lrf) 'lrf': -4., # final LambdaLR learning rate = lr0 * (10 ** lrf)
@ -375,7 +375,7 @@ if __name__ == '__main__':
x = np.loadtxt('evolve.txt', ndmin=2) x = np.loadtxt('evolve.txt', ndmin=2)
x = x[fitness(x).argmax()] # select best fitness hyps x = x[fitness(x).argmax()] # select best fitness hyps
for i, k in enumerate(hyp.keys()): for i, k in enumerate(hyp.keys()):
hyp[k] = x[i + 5] hyp[k] = x[i + 7]
# Mutate # Mutate
init_seeds(seed=int(time.time())) init_seeds(seed=int(time.time()))