Spaces:
Runtime error
Runtime error
Update detect_strongsort.py
Browse files- 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 |
|