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

updates

This commit is contained in:
Glenn Jocher 2019-12-21 20:17:56 -08:00
parent efb3768fff
commit f00de54546
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
train.py
View File

@ -25,7 +25,7 @@ results_file = 'results.txt'
hyp = {'giou': 3.54, # giou loss gain hyp = {'giou': 3.54, # giou loss gain
'cls': 37.4, # cls loss gain 'cls': 37.4, # cls loss gain
'cls_pw': 1.0, # cls BCELoss positive_weight 'cls_pw': 1.0, # cls BCELoss positive_weight
'obj': 64.3, # obj loss gain (*=img_size/416 if img_size != 416) 'obj': 49.5, # obj loss gain (*=img_size/320 if img_size != 320)
'obj_pw': 1.0, # obj BCELoss positive_weight 'obj_pw': 1.0, # obj BCELoss positive_weight
'iou_t': 0.225, # iou training threshold 'iou_t': 0.225, # iou training threshold
'lr0': 0.00579, # initial learning rate (SGD=1E-3, Adam=9E-5) 'lr0': 0.00579, # initial learning rate (SGD=1E-3, Adam=9E-5)
@ -445,8 +445,8 @@ if __name__ == '__main__':
if device.type == 'cpu': if device.type == 'cpu':
mixed_precision = False mixed_precision = False
# scale hyp['obj'] by img_size (evolved at 416) # scale hyp['obj'] by img_size (evolved at 320)
hyp['obj'] *= opt.img_size / 416. hyp['obj'] *= opt.img_size / 320.
tb_writer = None tb_writer = None
if not opt.evolve: # Train normally if not opt.evolve: # Train normally