From f3ce0d71363f08d70210a1cd6238d886414605d7 Mon Sep 17 00:00:00 2001 From: acai66 <1779864536@qq.com> Date: Thu, 18 Nov 2021 18:15:32 +0800 Subject: [PATCH] Turn down angle BCELoss positive_weight. --- data/hyps/hyp.finetune.yaml | 4 ++-- data/hyps/hyp.finetune_objects365.yaml | 4 ++-- data/hyps/hyp.scratch-high.yaml | 4 ++-- data/hyps/hyp.scratch-low.yaml | 2 ++ data/hyps/hyp.scratch-med.yaml | 2 ++ data/hyps/hyp.scratch.yaml | 4 ++-- 6 files changed, 12 insertions(+), 8 deletions(-) diff --git a/data/hyps/hyp.finetune.yaml b/data/hyps/hyp.finetune.yaml index ece1a5c..f589b49 100644 --- 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 index a6cc62f..20ab04b 100644 --- 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 index 2a113e5..6b923dc 100644 --- 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 index b9ef1d5..58d9138 100644 --- 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 index d6867d7..3f964f6 100644 --- 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 index f8a9449..8b428c1 100644 --- 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)