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

updates

This commit is contained in:
Glenn Jocher 2019-10-25 11:04:10 -05:00
parent b3b4ff4107
commit 8d1ab548c1
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

@ -22,9 +22,9 @@ results_file = 'results.txt'
# Hyperparameters (k-series, 53.3 mAP yolov3-spp-320) https://github.com/ultralytics/yolov3/issues/310 # Hyperparameters (k-series, 53.3 mAP yolov3-spp-320) https://github.com/ultralytics/yolov3/issues/310
hyp = {'giou': 3.31, # giou loss gain hyp = {'giou': 3.31, # giou loss gain
'cls': 42.4, # cls loss gain (CE=~1.0, uCE=~20) 'cls': 42.4, # cls loss gain
'cls_pw': 1.0, # cls BCELoss positive_weight 'cls_pw': 1.0, # cls BCELoss positive_weight
'obj': 40.0, # obj loss gain (*=80 for uBCE with 80 classes) 'obj': 40.0, # obj loss gain
'obj_pw': 1.0, # obj BCELoss positive_weight 'obj_pw': 1.0, # obj BCELoss positive_weight
'iou_t': 0.213, # iou training threshold 'iou_t': 0.213, # iou training threshold
'lr0': 0.00261, # initial learning rate (SGD=1E-3, Adam=9E-5) 'lr0': 0.00261, # initial learning rate (SGD=1E-3, Adam=9E-5)