Unverified 提交 d48a34dc authored 作者: Michael Heilig's avatar Michael Heilig 提交者: GitHub

bug fix: switched rows and cols for correct detections in confusion matrix (#2883)

上级 5f7d39fe
...@@ -145,7 +145,7 @@ class ConfusionMatrix: ...@@ -145,7 +145,7 @@ class ConfusionMatrix:
for i, gc in enumerate(gt_classes): for i, gc in enumerate(gt_classes):
j = m0 == i j = m0 == i
if n and sum(j) == 1: if n and sum(j) == 1:
self.matrix[gc, detection_classes[m1[j]]] += 1 # correct self.matrix[detection_classes[m1[j]], gc] += 1 # correct
else: else:
self.matrix[self.nc, gc] += 1 # background FP self.matrix[self.nc, gc] += 1 # background FP
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论