Confusion matrix native image-space fix (#2046)

Make sure the labels and predictions are equally scaled on confusion_matrix.process_batch
This commit is contained in:
ramonhollands authored and GitHub committed 2021-01-26 13:39:19 -08:00
1 parent 8fcbe43423
commit d68afedb32
1 file changed
+1 -1
+1 -1
View File
@@ -178,7 +178,7 @@ def test(data,
tbox = xywh2xyxy(labels[:, 1:5])
scale_coords(img[si].shape[1:], tbox, shapes[si][0], shapes[si][1]) # native-space labels
if plots:
confusion_matrix.process_batch(pred, torch.cat((labels[:, 0:1], tbox), 1))
confusion_matrix.process_batch(predn, torch.cat((labels[:, 0:1], tbox), 1))
# Per target class
for cls in torch.unique(tcls_tensor):