Apply mean and std normalization over input image.
(self, image: torch.Tensor)
| 34 | self.register_buffer("std_inv", 1.0 / std) |
| 35 | |
| 36 | def forward(self, image: torch.Tensor) -> torch.Tensor: |
| 37 | """Apply mean and std normalization over input image.""" |
| 38 | return (image - self.mean) * self.std_inv |
| 39 | |
| 40 | |
| 41 | class AffineRangeNormalizer(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected