From 8f480ead6a244068ed9696cdbe7ec23b3585e64e Mon Sep 17 00:00:00 2001
From: LinasKo <linkondrackis@gmail.com>
Date: Mon, 19 Aug 2024 00:52:58 +0300
Subject: [PATCH] Fix regression: OBB annotator

---
 supervision/annotators/core.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/supervision/annotators/core.py b/supervision/annotators/core.py
index dd628772..637dd9d8 100644
--- a/supervision/annotators/core.py
+++ b/supervision/annotators/core.py
@@ -262,9 +262,10 @@ class OrientedBoxAnnotator(BaseAnnotator):
         assert isinstance(scene, np.ndarray)
         if detections.data is None or ORIENTED_BOX_COORDINATES not in detections.data:
             return scene
+        obb_boxes = np.array(detections.data[ORIENTED_BOX_COORDINATES]).astype(int)
 
         for detection_idx in range(len(detections)):
-            bbox = detections.xyxy[detection_idx].astype(int)
+            obb = obb_boxes[detection_idx]
             color = resolve_color(
                 color=self.color,
                 detections=detections,
@@ -274,7 +275,7 @@ class OrientedBoxAnnotator(BaseAnnotator):
                 else custom_color_lookup,
             )
 
-            cv2.drawContours(scene, [bbox], 0, color.as_bgr(), self.thickness)
+            cv2.drawContours(scene, [obb], 0, color.as_bgr(), self.thickness)
 
         return scene