glenn-jocher commited on
Commit
1309400
·
1 Parent(s): fb05681

Update hyp.finetune.yaml on VOC results

Browse files
Files changed (1) hide show
  1. data/hyp.finetune.yaml +8 -8
data/hyp.finetune.yaml CHANGED
@@ -4,12 +4,12 @@
4
 
5
 
6
  lr0: 0.01 # initial learning rate (SGD=1E-2, Adam=1E-3)
7
- momentum: 0.937 # SGD momentum/Adam beta1
8
  weight_decay: 0.0005 # optimizer weight decay 5e-4
9
  giou: 0.05 # GIoU loss gain
10
- cls: 0.5 # cls loss gain
11
  cls_pw: 1.0 # cls BCELoss positive_weight
12
- obj: 1.0 # obj loss gain (scale with pixels)
13
  obj_pw: 1.0 # obj BCELoss positive_weight
14
  iou_t: 0.20 # IoU training threshold
15
  anchor_t: 4.0 # anchor-multiple threshold
@@ -17,11 +17,11 @@ fl_gamma: 0.0 # focal loss gamma (efficientDet default gamma=1.5)
17
  hsv_h: 0.015 # image HSV-Hue augmentation (fraction)
18
  hsv_s: 0.7 # image HSV-Saturation augmentation (fraction)
19
  hsv_v: 0.4 # image HSV-Value augmentation (fraction)
20
- degrees: 0.0 # image rotation (+/- deg)
21
  translate: 0.1 # image translation (+/- fraction)
22
- scale: 0.5 # image scale (+/- gain)
23
- shear: 0.0 # image shear (+/- deg)
24
  perspective: 0.0 # image perspective (+/- fraction), range 0-0.001
25
- flipud: 0.0 # image flip up-down (probability)
26
  fliplr: 0.5 # image flip left-right (probability)
27
- mixup: 0.0 # image mixup (probability)
 
4
 
5
 
6
  lr0: 0.01 # initial learning rate (SGD=1E-2, Adam=1E-3)
7
+ momentum: 0.94 # SGD momentum/Adam beta1
8
  weight_decay: 0.0005 # optimizer weight decay 5e-4
9
  giou: 0.05 # GIoU loss gain
10
+ cls: 0.2 # cls loss gain
11
  cls_pw: 1.0 # cls BCELoss positive_weight
12
+ obj: 0.3 # obj loss gain (scale with pixels)
13
  obj_pw: 1.0 # obj BCELoss positive_weight
14
  iou_t: 0.20 # IoU training threshold
15
  anchor_t: 4.0 # anchor-multiple threshold
 
17
  hsv_h: 0.015 # image HSV-Hue augmentation (fraction)
18
  hsv_s: 0.7 # image HSV-Saturation augmentation (fraction)
19
  hsv_v: 0.4 # image HSV-Value augmentation (fraction)
20
+ degrees: 1.0 # image rotation (+/- deg)
21
  translate: 0.1 # image translation (+/- fraction)
22
+ scale: 0.6 # image scale (+/- gain)
23
+ shear: 1.0 # image shear (+/- deg)
24
  perspective: 0.0 # image perspective (+/- fraction), range 0-0.001
25
+ flipud: 0.01 # image flip up-down (probability)
26
  fliplr: 0.5 # image flip left-right (probability)
27
+ mixup: 0.2 # image mixup (probability)