Spaces:
Runtime error
Runtime error
Change pred[0][1] to pred[0][0]
Browse files- 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][
|
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 |
|