Resets the internal evaluation result to initial state.
(self)
| 63 | return pixAcc, mIoU |
| 64 | |
| 65 | def reset(self): |
| 66 | """Resets the internal evaluation result to initial state.""" |
| 67 | self.total_inter = torch.zeros(self.nclass) |
| 68 | self.total_union = torch.zeros(self.nclass) |
| 69 | self.total_correct = 0 |
| 70 | self.total_label = 0 |
| 71 | |
| 72 | |
| 73 | # pytorch version |