Update loss.py

6 months ago · 83c967e1d6
parent 8d8ef9c479
commit 83c967e1d6
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/utils/loss.py
+++ b/ultralytics/utils/loss.py
@ -221,7 +221,7 @@ class v8DetectionLoss:
        dfl_conf = (dfl_conf.amax(-1).mean(-1) + dfl_conf.amax(-1).amin(-1)) / 2

        _, target_bboxes, target_scores, fg_mask, _ = self.assigner(
-            pred_scores.detach().sigmoid() * 0.4 + dfl_conf.unsqueeze(-1) * 0.6,
+            pred_scores.detach().sigmoid() * 0.7 + dfl_conf.unsqueeze(-1) * 0.3,
            # pred_scores.detach().sigmoid(),
            (pred_bboxes.detach() * stride_tensor).type(gt_bboxes.dtype),
            anchor_points * stride_tensor,