From 209807058a6d0bf2b5c898b5d5878fa28dced39f Mon Sep 17 00:00:00 2001 From: luanhailiang Date: Thu, 21 Sep 2023 21:50:38 +0800 Subject: [PATCH] // --- app.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/app.py b/app.py index 716f67f..022e7ac 100644 --- a/app.py +++ b/app.py @@ -79,9 +79,9 @@ def worker(id,source,region=None,stream=False): result = model.track(frame,show=False,stream=False,persist=True,device=int(id)%4) result = result[0] detections = sv.Detections.from_yolov8(result) - # if result.boxes.id is not None: - # detections.tracker_id = result.boxes.id.cpu().numpy().astype(int) - # detections = detections[(detections.tracker_id is not None)] + if result.boxes.id is not None: + detections.tracker_id = result.boxes.id.cpu().numpy().astype(int) + detections = detections[(detections.tracker_id is not None)] count += 1 names = {}