HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Minor fixes (#162)

Browse source
This commit is contained in:
Laughing 2023-01-09 20:41:34 +08:00 • committed by GitHub
parent 9a2f67b3b4
commit 20fee4100c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 15 additions and 53 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ultralytics/yolo/v8/detect/train.py
View file

@ -186,9 +186,9 @@ class Loss:
loss[0], loss[2] = self.bbox_loss(pred_distri, pred_bboxes, anchor_points, target_bboxes, target_scores,
target_scores_sum, fg_mask)
loss[0] *= 7.5 # box gain
loss[1] *= 0.5 # cls gain
loss[2] *= 1.5 # dfl gain
loss[0] *= self.hyp.box # box gain
loss[1] *= self.hyp.cls # cls gain
loss[2] *= self.hyp.dfl # dfl gain
return loss.sum() * batch_size, loss.detach() # loss(box, cls, dfl)