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-21 00:21:36 +02:00
parent 291def2e77
commit 9dfce6e5de
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -39,8 +39,8 @@ hyp = {'giou': 1.582, # giou loss gain
'wh': 0.1857, # wh loss gain 'wh': 0.1857, # wh loss gain
'cls': 27.76, # cls loss gain (CE=~1.0, uCE=~20) 'cls': 27.76, # cls loss gain (CE=~1.0, uCE=~20)
'cls_pw': 1.446, # cls BCELoss positive_weight 'cls_pw': 1.446, # cls BCELoss positive_weight
'obj': 21.35, # obj loss gain 'obj': 21.35, # obj loss gain (*=80 for uBCE with 80 classes)
'obj_pw': 3.941, # obj BCELoss positive_weight (*=80 for uBCE with 80 classes) 'obj_pw': 3.941, # obj BCELoss positive_weight
'iou_t': 0.2635, # iou training threshold 'iou_t': 0.2635, # iou training threshold
'lr0': 0.002324, # initial learning rate 'lr0': 0.002324, # initial learning rate
'lrf': -4., # final LambdaLR learning rate = lr0 * (10 ** lrf) 'lrf': -4., # final LambdaLR learning rate = lr0 * (10 ** lrf)

2
utils/utils.py
View File

@ -367,7 +367,7 @@ def compute_loss(p, targets, model): # predictions, targets, model
t = torch.zeros_like(pi[..., 5:]) # targets t = torch.zeros_like(pi[..., 5:]) # targets
if nb: if nb:
t[b, a, gj, gi, tcls[i]] = 1.0 t[b, a, gj, gi, tcls[i]] = 1.0
lcls += BCEcls(pi[..., 5:], t) lcls += BCEobj(pi[..., 5:], t)
lbox *= k * h['giou'] lbox *= k * h['giou']
lobj *= k * h['obj'] lobj *= k * h['obj']