ChristopherSTAN
commited on
Update train.py
Browse files
train.py
CHANGED
@@ -37,7 +37,7 @@ hyp = {'lr0': 0.01, # initial learning rate (SGD=1E-2, Adam=1E-3)
|
|
37 |
'obj_pw': 1.0, # obj BCELoss positive_weight
|
38 |
'iou_t': 0.20, # iou training threshold
|
39 |
'anchor_t': 4.0, # anchor-multiple threshold
|
40 |
-
'fl_gamma':
|
41 |
'hsv_h': 0.014, # image HSV-Hue augmentation (fraction)
|
42 |
'hsv_s': 0.68, # image HSV-Saturation augmentation (fraction)
|
43 |
'hsv_v': 0.36, # image HSV-Value augmentation (fraction)
|
|
|
37 |
'obj_pw': 1.0, # obj BCELoss positive_weight
|
38 |
'iou_t': 0.20, # iou training threshold
|
39 |
'anchor_t': 4.0, # anchor-multiple threshold
|
40 |
+
'fl_gamma': 1.5, # focal loss gamma (efficientDet default is gamma=1.5)
|
41 |
'hsv_h': 0.014, # image HSV-Hue augmentation (fraction)
|
42 |
'hsv_s': 0.68, # image HSV-Saturation augmentation (fraction)
|
43 |
'hsv_v': 0.36, # image HSV-Value augmentation (fraction)
|