ntsc207 commited on
Commit
a75c823
1 Parent(s): 0b99468

Update detect_strongsort.py

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