Spaces:

k20hcmus
/

FishEye8K

Runtime error

ntsc207 commited on May 12, 2024

Commit

d5dad45

verified ·

1 Parent(s): a4e7334

Update detect_strongsort.py

Files changed (1) hide show

detect_strongsort.py CHANGED Viewed

@@ -138,12 +138,11 @@ def run_strongsort(
     # Create as many strong sort instances as there are video sources
     strongsort_list = []
-    cpu = 'cpu'
     for i in range(bs):
         strongsort_list.append(
             StrongSORT(
                 strong_sort_weights,
-                cpu,
                 half,
                 max_dist=cfg.STRONGSORT.MAX_DIST,
                 max_iou_distance=cfg.STRONGSORT.MAX_IOU_DISTANCE,

     # Create as many strong sort instances as there are video sources
     strongsort_list = []
     for i in range(bs):
         strongsort_list.append(
             StrongSORT(
                 strong_sort_weights,
+                device,
                 half,
                 max_dist=cfg.STRONGSORT.MAX_DIST,
                 max_iou_distance=cfg.STRONGSORT.MAX_IOU_DISTANCE,