ntsc207 commited on
Commit
deab8af
·
verified ·
1 Parent(s): 5dd6bd8

Update detect_strongsort.py

Browse files
Files changed (1) hide show
  1. 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], [0,148,255], [0, 255, 10], [250, 247, 0], [235,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]