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

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
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