jasder
/
yolov5
1
0
Fork 0
Code Issues Pull Requests 1 Releases Wiki Activity

Update train.py

Browse Source
pull/1/head
Glenn Jocher 5 years ago committed by GitHub
parent a8d17ce712
commit 9705c15da1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
train.py
Unescape View File

@ -37,7 +37,7 @@ hyp = {'lr0': 0.01, # initial learning rate (SGD=1E-2, Adam=1E-3)
'obj_pw': 1.0, # obj BCELoss positive_weight 'obj_pw': 1.0, # obj BCELoss positive_weight
'iou_t': 0.20, # iou training threshold 'iou_t': 0.20, # iou training threshold
'anchor_t': 4.0, # anchor-multiple threshold 'anchor_t': 4.0, # anchor-multiple threshold
'fl_gamma': 1.5, # focal loss gamma (efficientDet default is gamma=1.5) 'fl_gamma': 0.0, # focal loss gamma (efficientDet default is gamma=1.5)
'hsv_h': 0.014, # image HSV-Hue augmentation (fraction) 'hsv_h': 0.014, # image HSV-Hue augmentation (fraction)
'hsv_s': 0.68, # image HSV-Saturation augmentation (fraction) 'hsv_s': 0.68, # image HSV-Saturation augmentation (fraction)
'hsv_v': 0.36, # image HSV-Value augmentation (fraction) 'hsv_v': 0.36, # image HSV-Value augmentation (fraction)

Write Preview
Loading…
Cancel
Save