Spaces:
Running
on
Zero
Running
on
Zero
Update detect_strongsort.py
Browse files- detect_strongsort.py +1 -2
detect_strongsort.py
CHANGED
@@ -137,13 +137,12 @@ def run_strongsort(
|
|
137 |
cfg.merge_from_file(config_strongsort)
|
138 |
|
139 |
# Create as many strong sort instances as there are video sources
|
140 |
-
gpu = '0'
|
141 |
strongsort_list = []
|
142 |
for i in range(bs):
|
143 |
strongsort_list.append(
|
144 |
StrongSORT(
|
145 |
strong_sort_weights,
|
146 |
-
|
147 |
half,
|
148 |
max_dist=cfg.STRONGSORT.MAX_DIST,
|
149 |
max_iou_distance=cfg.STRONGSORT.MAX_IOU_DISTANCE,
|
|
|
137 |
cfg.merge_from_file(config_strongsort)
|
138 |
|
139 |
# Create as many strong sort instances as there are video sources
|
|
|
140 |
strongsort_list = []
|
141 |
for i in range(bs):
|
142 |
strongsort_list.append(
|
143 |
StrongSORT(
|
144 |
strong_sort_weights,
|
145 |
+
device,
|
146 |
half,
|
147 |
max_dist=cfg.STRONGSORT.MAX_DIST,
|
148 |
max_iou_distance=cfg.STRONGSORT.MAX_IOU_DISTANCE,
|