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

Code refactor https://ultralytics.com/actions (#13844)

Browse source
This commit is contained in:
Glenn Jocher 2024-06-21 01:10:52 +02:00 • committed by GitHub
parent ffb46fd7fb
commit ecfcdf12c9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 21 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

3
ultralytics/nn/modules/head.py
View file

@ -110,8 +110,7 @@ class Detect(nn.Module):
else:
dbox = self.decode_bboxes(self.dfl(box), self.anchors.unsqueeze(0)) * self.strides
y = torch.cat((dbox, cls.sigmoid()), 1)
return y
return torch.cat((dbox, cls.sigmoid()), 1)
def bias_init(self):
"""Initialize Detect() biases, WARNING: requires stride availability."""