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