xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-06-04 06:24:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Muhammad Rizwan Munawar 2023-10-29 22:53:09 +05:00 committed by GitHub
parent e58db228c2
commit ae6fa2fb02
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 23 deletions

2
examples/YOLOv8-Region-Counter/yolov8_region_counter.py

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