Update detect_deepsort.py

detect_deepsort.py

@@ -204,6 +204,7 @@ def run_deepsort(
 
         # 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)
 
         # Second-stage classifier (optional)