Chiebot-Mirror
/
ultralytics
mirror of https://github.com/ultralytics/ultralytics.git
9
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Update loss.py

Browse Source
exp-b
Laughing-q 3 months ago
parent 83c967e1d6
commit 3d81d9bba5
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      ultralytics/utils/loss.py

2
ultralytics/utils/loss.py
Unescape View File

@ -221,7 +221,7 @@ class v8DetectionLoss:
dfl_conf = (dfl_conf.amax(-1).mean(-1) + dfl_conf.amax(-1).amin(-1)) / 2 dfl_conf = (dfl_conf.amax(-1).mean(-1) + dfl_conf.amax(-1).amin(-1)) / 2
_, target_bboxes, target_scores, fg_mask, _ = self.assigner( _, 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_scores.detach().sigmoid(),
(pred_bboxes.detach() * stride_tensor).type(gt_bboxes.dtype), (pred_bboxes.detach() * stride_tensor).type(gt_bboxes.dtype),
anchor_points * stride_tensor, anchor_points * stride_tensor,

Write Preview
Loading…
Cancel
Save