,device=int(id)%4

2023-08-14 17:02:10 +08:00 · 2023-08-14 17:02:10 +08:00 · 8fdc5c3b64
parent af4ed177f6
commit 8fdc5c3b64
1 changed files with 1 additions and 1 deletions
--- a/app.py
+++ b/app.py
@ -60,7 +60,7 @@ def worker(id,source,region=None,stream=False):
            if not ret:
                print("Error read video stream.")
                break
-            result = model.track(frame,show=False,stream=False,persist=True)
+            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: