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

Fix typo in docstring in metrics.py (#15842)

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Jan Knobloch 2024-08-27 21:05:33 +02:00 • committed by GitHub
parent db047e4e88
commit b1a80f4d98
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/utils/metrics.py
View file

@ -178,7 +178,7 @@ def _get_covariance_matrix(boxes):
boxes (torch.Tensor): A tensor of shape (N, 5) representing rotated bounding boxes, with xywhr format. boxes (torch.Tensor): A tensor of shape (N, 5) representing rotated bounding boxes, with xywhr format.
Returns: Returns:
(torch.Tensor): Covariance metrixs corresponding to original rotated bounding boxes. (torch.Tensor): Covariance matrices corresponding to original rotated bounding boxes.
""" """
# Gaussian bounding boxes, ignore the center points (the first two columns) because they are not needed here. # Gaussian bounding boxes, ignore the center points (the first two columns) because they are not needed here.
gbbs = torch.cat((boxes[:, 2:4].pow(2) / 12, boxes[:, 4:]), dim=-1) gbbs = torch.cat((boxes[:, 2:4].pow(2) / 12, boxes[:, 4:]), dim=-1)