Update detect_strongsort.py

detect_strongsort.py

@@ -185,12 +185,13 @@ def run_strongsort(
         with dt[1]:
             visualize = increment_path(save_dir / Path(path).stem, mkdir=True) if visualize else False
             pred = model(im, augment=augment, visualize=visualize)
-            pred = pred[0][1]
+            # pred = pred[0][1]
         t3 = time_sync()
         sdt[1] += t3 - t2
 
         # Apply NMS
         with dt[2]:
+            pred = pred[0][1] if isinstance(pred[0], list) else pred[0]  # single model or ensemble
             pred = non_max_suppression(pred, conf_thres, iou_thres, classes, agnostic_nms, max_det=max_det)
         sdt[2] += time_sync() - t3