Spaces:
Running
on
Zero
Running
on
Zero
Update detect_strongsort.py
Browse files- detect_strongsort.py +1 -1
detect_strongsort.py
CHANGED
@@ -161,7 +161,7 @@ def run_strongsort(
|
|
161 |
strongsort_list[i].model.warmup()
|
162 |
outputs = [None] * bs
|
163 |
|
164 |
-
colors = [[0, 0, 255], [
|
165 |
#250, 247, 0
|
166 |
# Run tracking
|
167 |
model.warmup(imgsz=(1 if pt or model.triton else bs, 3, *imgsz)) # warmup
|
|
|
161 |
strongsort_list[i].model.warmup()
|
162 |
outputs = [None] * bs
|
163 |
|
164 |
+
colors = [[0, 0, 255], [255, 148, 0], [0, 255, 10], [0, 247, 250], [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
|