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

Fix xyxyxyxyn calculation, swap axis (#8249)

Browse source
This commit is contained in:
Johannes Kaisinger 2024-02-17 01:30:04 +01:00 • committed by GitHub
parent 2f22ff4ae6
commit 8aa73c4470
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/engine/results.py
View file

@ -710,8 +710,8 @@ 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[0] xyxyxyxyn[..., 0] /= self.orig_shape[1]
xyxyxyxyn[..., 1] /= self.orig_shape[1] xyxyxyxyn[..., 1] /= self.orig_shape[0]
return xyxyxyxyn return xyxyxyxyn
@property @property