ntsc207 commited on
Commit
9ad66ab
·
verified ·
1 Parent(s): 5c10599

Change pred[0][1] to pred[0][0]

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