HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

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

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
Download patch file Download diff file Expand all files Collapse all files

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

@ -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()