Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-04-02 14:02:35 +02:00
parent c9328f663f
commit e3781460f8
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test.py
View File

@ -89,10 +89,7 @@ def test(
'score': float(d[4]) 'score': float(d[4])
}) })
# If no labels add number of detections as incorrect if len(labels):
if len(labels) == 0:
correct.extend([0] * len(pred))
else:
# Extract target boxes as (x1, y1, x2, y2) # Extract target boxes as (x1, y1, x2, y2)
tbox = xywh2xyxy(labels[:, 1:5]) * img_size # target boxes tbox = xywh2xyxy(labels[:, 1:5]) * img_size # target boxes
tcls = labels[:, 0] # target classes tcls = labels[:, 0] # target classes
@ -111,6 +108,9 @@ def test(
detected.append(bi) detected.append(bi)
else: else:
correct.append(0) correct.append(0)
else:
# If no labels add number of detections as incorrect
correct.extend([0] * len(pred))
# Append Statistics (correct, conf, pcls, tcls) # Append Statistics (correct, conf, pcls, tcls)
stats.append((correct, pred[:, 4].cpu(), pred[:, 6].cpu(), tcls.cpu())) stats.append((correct, pred[:, 4].cpu(), pred[:, 6].cpu(), tcls.cpu()))