Turn down angle BCELoss positive_weight.

acai66 · Nov 18, 2021 · f3ce0d7 · f3ce0d7
1 parent a4b4dff
commit f3ce0d7
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 8 deletions.
diff --git a/data/hyps/hyp.finetune.yaml b/data/hyps/hyp.finetune.yaml
@@ -20,8 +20,8 @@ cls: 0.243
 cls_pw: 0.631
 obj: 0.301
 obj_pw: 0.911
-angle: 0.266
-angle_pw: 0.333  # angle BCELoss positive_weight
+angle: 0.03
+angle_pw: 0.05  # angle BCELoss positive_weight
 iou_t: 0.2
 anchor_t: 2.91
 # anchors: 3.63

diff --git a/data/hyps/hyp.finetune_objects365.yaml b/data/hyps/hyp.finetune_objects365.yaml
@@ -12,8 +12,8 @@ cls: 0.299
 cls_pw: 0.825
 obj: 0.632
 obj_pw: 1.0
-angle: 0.266
-angle_pw: 0.333  # angle BCELoss positive_weight
+angle: 0.03
+angle_pw: 0.05  # angle BCELoss positive_weight
 iou_t: 0.2
 anchor_t: 3.44
 anchors: 3.2

diff --git a/data/hyps/hyp.scratch-high.yaml b/data/hyps/hyp.scratch-high.yaml
@@ -15,8 +15,8 @@ cls: 0.3  # cls loss gain
 cls_pw: 1.0  # cls BCELoss positive_weight
 obj: 0.7  # obj loss gain (scale with pixels)
 obj_pw: 1.0  # obj BCELoss positive_weight
-angle: 0.266
-angle_pw: 0.333  # angle BCELoss positive_weight
+angle: 0.03
+angle_pw: 0.05  # angle BCELoss positive_weight
 iou_t: 0.20  # IoU training threshold
 anchor_t: 4.0  # anchor-multiple threshold
 # anchors: 3  # anchors per output layer (0 to ignore)

diff --git a/data/hyps/hyp.scratch-low.yaml b/data/hyps/hyp.scratch-low.yaml
@@ -15,6 +15,8 @@ cls: 0.5  # cls loss gain
 cls_pw: 1.0  # cls BCELoss positive_weight
 obj: 1.0  # obj loss gain (scale with pixels)
 obj_pw: 1.0  # obj BCELoss positive_weight
+angle: 0.03
+angle_pw: 0.05  # angle BCELoss positive_weight
 iou_t: 0.20  # IoU training threshold
 anchor_t: 4.0  # anchor-multiple threshold
 # anchors: 3  # anchors per output layer (0 to ignore)

diff --git a/data/hyps/hyp.scratch-med.yaml b/data/hyps/hyp.scratch-med.yaml
@@ -15,6 +15,8 @@ cls: 0.3  # cls loss gain
 cls_pw: 1.0  # cls BCELoss positive_weight
 obj: 0.7  # obj loss gain (scale with pixels)
 obj_pw: 1.0  # obj BCELoss positive_weight
+angle: 0.03
+angle_pw: 0.05  # angle BCELoss positive_weight
 iou_t: 0.20  # IoU training threshold
 anchor_t: 4.0  # anchor-multiple threshold
 # anchors: 3  # anchors per output layer (0 to ignore)

diff --git a/data/hyps/hyp.scratch.yaml b/data/hyps/hyp.scratch.yaml
@@ -15,8 +15,8 @@ cls: 0.5  # cls loss gain
 cls_pw: 1.0  # cls BCELoss positive_weight
 obj: 1.0  # obj loss gain (scale with pixels)
 obj_pw: 1.0  # obj BCELoss positive_weight
-angle: 0.266
-angle_pw: 0.333  # angle BCELoss positive_weight
+angle: 0.03
+angle_pw: 0.05  # angle BCELoss positive_weight
 iou_t: 0.20  # IoU training threshold
 anchor_t: 4.0  # anchor-multiple threshold
 # anchors: 3  # anchors per output layer (0 to ignore)