| 57 | |
| 58 | |
| 59 | class DeNormalize(object): |
| 60 | def __init__(self, mean, std): |
| 61 | self.mean = mean |
| 62 | self.std = std |
| 63 | |
| 64 | def __call__(self, tensor): |
| 65 | for t, m, s in zip(tensor, self.mean, self.std): |
| 66 | t.mul_(s).add_(m) |
| 67 | return tensor |
| 68 | |
| 69 | |
| 70 | class MaskToTensor(object): |
nothing calls this directly
no outgoing calls
no test coverage detected