ChristopherSTAN
GitHub
4 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
train.py
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 | |||
| 'iou_t': 0.20, # iou training threshold | |||
| 'anchor_t': 4.0, # anchor-multiple threshold | |||
| 'fl_gamma': 0.0, # focal loss gamma (efficientDet default is gamma=1.5) | |||
| 'fl_gamma': 1.5, # focal loss gamma (efficientDet default is gamma=1.5) | |||
| 'hsv_h': 0.014, # image HSV-Hue augmentation (fraction) | |||
| 'hsv_s': 0.68, # image HSV-Saturation augmentation (fraction) | |||
| 'hsv_v': 0.36, # image HSV-Value augmentation (fraction) | |||
Loading…