|
|
|
@@ -121,7 +121,7 @@ def test(data, |
|
|
|
out = non_max_suppression(out, conf_thres=conf_thres, iou_thres=iou_thres, labels=lb, multi_label=True) |
|
|
|
t1 += time_synchronized() - t |
|
|
|
|
|
|
|
# Statistics per image |
|
|
|
# Statistics per image and |
|
|
|
for si, pred in enumerate(out): |
|
|
|
labels = targets[targets[:, 0] == si, 1:] |
|
|
|
nl = len(labels) |
|
|
|
|