Fix tracks=None bug in YOLOv8-Region-Counting module (#5977)

2023-10-29 22:53:09 +05:00 · 2023-10-29 22:53:09 +05:00 · ae6fa2fb02
commit ae6fa2fb02
parent e58db228c2
1 changed files with 25 additions and 23 deletions
--- a/examples/YOLOv8-Region-Counter/yolov8_region_counter.py
+++ b/examples/YOLOv8-Region-Counter/yolov8_region_counter.py
@ -120,6 +120,8 @@ def run(

        # Extract the results
        results = model.track(frame, persist=True)
+
+        if results[0].boxes.id is not None:
            boxes = results[0].boxes.xywh.cpu()
            track_ids = results[0].boxes.id.int().cpu().tolist()
            clss = results[0].boxes.cls.cpu().tolist()