crop

factories.py

@@ -472,7 +472,7 @@ class Metric():
         elif self.name == "LPIPS":
             self.metric = dinv.loss.metric.LPIPS(device=device_str)
 
-    def __call__(self, x_net: torch.Tensor, x: torch.Tensor, *args, **kwargs) -> torch.Tensor:
+    def __call__(self, x_net: torch.Tensor, x: torch.Tensor, *args, crop=31, **kwargs) -> torch.Tensor:
         # it may happen that x_net and x do not have the same size, in which case we take the minimum size of both
         if x_net.shape[-1] != x.shape[-1]:
             min_size = min(x_net.shape[-1], x.shape[-1])
@@ -483,7 +483,7 @@ class Metric():
         else:
             x_net_crop = x_net
             x_crop = x
-        return self.metric(x_net_crop, x_crop)
+        return self.metric(x_net_crop[..., crop:-crop, crop:-crop], x_crop[..., crop:-crop, crop:-crop])
 
     @classmethod
     def get_list_metrics(cls, metric_names: List[str], device_str: str = "cpu") -> List["Metric"]: