feat: upscale before lpips inference

junhsss · Mar 24, 2023 · 2e0ec57 · 2e0ec57
1 parent 6dad56b
commit 2e0ec57
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/consistency/loss.py b/consistency/loss.py
@@ -42,8 +42,12 @@ def _append_net_type(net_type: str):
         self.l1_weight = l1_weight
 
     def forward(self, input, target):
+        upscaled_input = F.interpolate(input, (224, 224), mode="bilinear")
+        upscaled_target = F.interpolate(target, (224, 224), mode="bilinear")
+
         lpips_loss = sum(
-            _lpips_loss(input, target) for _lpips_loss in self.lpips_losses
+            _lpips_loss(upscaled_input, upscaled_target)
+            for _lpips_loss in self.lpips_losses
         )
 
         return lpips_loss + self.l1_weight * F.l1_loss(input, target)

diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 from setuptools import find_packages, setup
 
-__version__ = "0.2.3"
+__version__ = "0.2.4"
 
 setup(
     name="consistency",