diff --git a/ultralytics/utils/loss.py b/ultralytics/utils/loss.py index b034ff5fd6..ddd1f0ec24 100644 --- 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.7 + dfl_conf.unsqueeze(-1) * 0.3, + pred_scores.detach().sigmoid() * 0.8 + dfl_conf.unsqueeze(-1) * 0.2, # pred_scores.detach().sigmoid(), (pred_bboxes.detach() * stride_tensor).type(gt_bboxes.dtype), anchor_points * stride_tensor,