Fix Classify confusion matrix (#3241)
This commit is contained in:
parent
9c44560ec4
commit
9d38d2b044
1 changed files with 1 additions and 1 deletions
|
|
@ -204,7 +204,7 @@ class ConfusionMatrix:
|
|||
"""
|
||||
preds, targets = torch.cat(preds)[:, 0], torch.cat(targets)
|
||||
for p, t in zip(preds.cpu().numpy(), targets.cpu().numpy()):
|
||||
self.matrix[t][p] += 1
|
||||
self.matrix[p][t] += 1
|
||||
|
||||
def process_batch(self, detections, labels):
|
||||
"""
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue