Browse Source

Update test.py

master
bailulu 3 years ago
parent
commit
de93f2d48e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      test.py

+ 1
- 1
test.py View File

@@ -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)


Loading…
Cancel
Save