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

Add YOLOv9 Docs page (#8478)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-02-26 23:09:07 +01:00 • committed by GitHub
parent 2d75f72598
commit 0efbd2d7cf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 120 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

7
ultralytics/utils/loss.py
View file

@ -597,7 +597,12 @@ class v8ClassificationLoss:
class v8OBBLoss(v8DetectionLoss):
def __init__(self, model): # model must be de-paralleled
def __init__(self, model):
"""
Initializes v8OBBLoss with model, assigner, and rotated bbox loss.
Note model must be de-paralleled.
"""
super().__init__(model)
self.assigner = RotatedTaskAlignedAssigner(topk=10, num_classes=self.nc, alpha=0.5, beta=6.0)
self.bbox_loss = RotatedBboxLoss(self.reg_max - 1, use_dfl=self.use_dfl).to(self.device)