Spaces:
Running
on
Zero
Running
on
Zero
Update detect.py
Browse files
detect.py
CHANGED
@@ -100,7 +100,7 @@ def run(
|
|
100 |
|
101 |
# NMS
|
102 |
with dt[2]:
|
103 |
-
pred = pred[0][1] if isinstance(pred[0], list) else pred[0] # single model or ensemble
|
104 |
pred = non_max_suppression(pred, conf_thres, iou_thres, classes, agnostic_nms, max_det=max_det)
|
105 |
|
106 |
|
|
|
100 |
|
101 |
# NMS
|
102 |
with dt[2]:
|
103 |
+
# pred = pred[0][1] if isinstance(pred[0], list) else pred[0] # single model or ensemble
|
104 |
pred = non_max_suppression(pred, conf_thres, iou_thres, classes, agnostic_nms, max_det=max_det)
|
105 |
|
106 |
|