Spaces:
Running
on
Zero
Running
on
Zero
Update detect_strongsort.py
Browse files- detect_strongsort.py +2 -2
detect_strongsort.py
CHANGED
@@ -161,8 +161,8 @@ def run_strongsort(
|
|
161 |
strongsort_list[i].model.warmup()
|
162 |
outputs = [None] * bs
|
163 |
|
164 |
-
colors = [[0, 0, 255], [
|
165 |
-
|
166 |
# Run tracking
|
167 |
model.warmup(imgsz=(1 if pt or model.triton else bs, 3, *imgsz)) # warmup
|
168 |
seen, windows, dt,sdt = 0, [], (Profile(), Profile(), Profile(), Profile()),[0.0, 0.0, 0.0, 0.0]
|
|
|
161 |
strongsort_list[i].model.warmup()
|
162 |
outputs = [None] * bs
|
163 |
|
164 |
+
colors = [[0, 0, 255], [250, 247, 0], [0, 255, 10], [0,148,255], [235,0,255]]
|
165 |
+
#250, 247, 0
|
166 |
# Run tracking
|
167 |
model.warmup(imgsz=(1 if pt or model.triton else bs, 3, *imgsz)) # warmup
|
168 |
seen, windows, dt,sdt = 0, [], (Profile(), Profile(), Profile(), Profile()),[0.0, 0.0, 0.0, 0.0]
|