tuansunday05 commited on
Commit
4512b7d
·
verified ·
1 Parent(s): 07c482c

Update detect_strongsort.py

Browse files
Files changed (1) hide show
  1. detect_strongsort.py +1 -1
detect_strongsort.py CHANGED
@@ -179,7 +179,7 @@ def run_strongsort(
179
  with dt[1]:
180
  visualize = increment_path(save_dir / Path(path).stem, mkdir=True) if visualize else False
181
  pred = model(im, augment=augment, visualize=visualize)
182
- pred = pred[0][1]
183
  t3 = time_sync()
184
  sdt[1] += t3 - t2
185
 
 
179
  with dt[1]:
180
  visualize = increment_path(save_dir / Path(path).stem, mkdir=True) if visualize else False
181
  pred = model(im, augment=augment, visualize=visualize)
182
+ # pred = pred[0][1]
183
  t3 = time_sync()
184
  sdt[1] += t3 - t2
185