Skip to content

GitLab

Unverified Commit 622bb269 authored by Siddharth Singh's avatar Siddharth Singh Committed by GitHub
Browse files

Fix IoUs reported in segmentation references (#7916) 


Co-authored-by: default avatarNicolas Hug <nh.nicolas.hug@gmail.com>
parent 35677eca
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
references/segmentation/utils.py
View file @ 622bb269
...@@ -88,7 +88,7 @@ class ConfusionMatrix: ...@@ -88,7 +88,7 @@ class ConfusionMatrix:
return acc_global, acc, iu return acc_global, acc, iu
def reduce_from_all_processes(self): def reduce_from_all_processes(self):
reduce_across_processes(self.mat) self.mat = reduce_across_processes(self.mat).to(torch.int64)
def __str__(self): def __str__(self):
acc_global, acc, iu = self.compute() acc_global, acc, iu = self.compute()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment