xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix xyxyxyxyn calculation in obb result (#8188)

Browse Source
This commit is contained in:
Johannes Kaisinger 2024-02-13 23:54:41 +01:00 committed by GitHub
parent 22feb682d4
commit f9e9cdf2c3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/engine/results.py
View File

@ -710,7 +710,7 @@ class OBB(BaseTensor):
def xyxyxyxyn(self): def xyxyxyxyn(self):
"""Return the boxes in xyxyxyxy format, (N, 4, 2).""" """Return the boxes in xyxyxyxy format, (N, 4, 2)."""
xyxyxyxyn = self.xyxyxyxy.clone() if isinstance(self.xyxyxyxy, torch.Tensor) else np.copy(self.xyxyxyxy) xyxyxyxyn = self.xyxyxyxy.clone() if isinstance(self.xyxyxyxy, torch.Tensor) else np.copy(self.xyxyxyxy)
xyxyxyxyn[..., 0] /= self.orig_shape[1] xyxyxyxyn[..., 0] /= self.orig_shape[0]
xyxyxyxyn[..., 1] /= self.orig_shape[1] xyxyxyxyn[..., 1] /= self.orig_shape[1]
return xyxyxyxyn return xyxyxyxyn