Fix xyxyxyxyn calculation, swap axis (#8249)

9 months ago · 8aa73c4470
parent 2f22ff4ae6
commit 8aa73c4470
1 changed files with 2 additions and 2 deletions
--- a/ultralytics/engine/results.py
+++ b/ultralytics/engine/results.py
@ -710,8 +710,8 @@ class OBB(BaseTensor):
    def xyxyxyxyn(self):
        """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[..., 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
    @property