Method __call__

script/feature/misc.py:297–309

Args: tensor (Tensor): Tensor image of size (C, H, W) to be normalized. Returns: Tensor: Normalized image.

(self, tensor)

        self.std = std

    def __call__(self, tensor):
        """
        Args:
            tensor (Tensor): Tensor image of size (C, H, W) to be normalized.
        Returns:
            Tensor: Normalized image.
        """
        batch_size = tensor.size(0)
        for i in range(batch_size):
            for t, m, s in zip(tensor[i], self.mean, self.std):
                t.mul_(s).add_(m)
                # The normalize code -> t.sub_(m).div_(s)
        return tensor

class L2Loss(nn.Module):
    """