From ba949830bfd232dccd3aa531f6e1d92583ad4dbd Mon Sep 17 00:00:00 2001 From: Muhammad Rizwan Munawar Date: Tue, 24 Sep 2024 03:40:05 +0500 Subject: [PATCH] `ultralytics 8.2.100` new YOLOv8-OBB object counting (#16437) Co-authored-by: Glenn Jocher --- ultralytics/__init__.py | 2 +- ultralytics/solutions/object_counter.py | 11 +++++++---- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index a1754a4875..85944a543c 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.2.99" +__version__ = "8.2.100" import os diff --git a/ultralytics/solutions/object_counter.py b/ultralytics/solutions/object_counter.py index ce6c1cb579..398629a8bc 100644 --- a/ultralytics/solutions/object_counter.py +++ b/ultralytics/solutions/object_counter.py @@ -127,10 +127,13 @@ class ObjectCounter: # Draw region or line annotator.draw_region(reg_pts=self.reg_pts, color=(104, 0, 123), thickness=self.tf * 2) - if tracks[0].boxes.id is not None: - boxes = tracks[0].boxes.xyxy.cpu() - clss = tracks[0].boxes.cls.cpu().tolist() - track_ids = tracks[0].boxes.id.int().cpu().tolist() + # Extract tracks for OBB or object detection + track_data = tracks[0].obb or tracks[0].boxes + + if track_data and track_data.id is not None: + boxes = track_data.xyxy.cpu() + clss = track_data.cls.cpu().tolist() + track_ids = track_data.id.int().cpu().tolist() # Extract tracks for box, track_id, cls in zip(boxes, track_ids, clss):