Method __init__

tools/test_reconstruction_hf.py:68–70

(self, device='cuda')

    """LPIPS metric wrapper."""

    def __init__(self, device='cuda'):
        self.device = device
        self.model = LPIPS_VTP().to(device).eval()

    def __call__(self, img1: torch.Tensor, img2: torch.Tensor) -> torch.Tensor:
        """Calculate LPIPS between two images.