(self, pred, label)
| 13 | super(SegmentationModuleBase, self).__init__() |
| 14 | |
| 15 | def pixel_acc(self, pred, label): |
| 16 | _, preds = torch.max(pred, dim=1) |
| 17 | valid = (label >= 0).long() |
| 18 | acc_sum = torch.sum(valid * (preds == label).long()) |
| 19 | pixel_sum = torch.sum(valid) |
| 20 | acc = acc_sum.float() / (pixel_sum.float() + 1e-10) |
| 21 | return acc |
| 22 | |
| 23 | |
| 24 | class SegmentationModule(SegmentationModuleBase): |