fix: incorrect resizing in adabins

samedii · Oct 15, 2022 · 9559c65 · 9559c65
1 parent 3847c96
commit 9559c65
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/perceptor/models/adabins_depth/adabins_depth.py b/perceptor/models/adabins_depth/adabins_depth.py
@@ -38,12 +38,12 @@ def forward(self, images: Tensor.dims("NCHW")) -> Tensor.dims("NCHW"):
         if image_area > MAX_ADABINS_AREA:
             scale = np.sqrt(MAX_ADABINS_AREA) / np.sqrt(image_area)
             images = transforms.resize(
-                images, out_shape=(int(w * scale), int(h * scale)), resample="lancsoz3"
+                images, out_shape=(int(h * scale), int(w * scale)), resample="lanczos3"
             )
         elif image_area < MIN_ADABINS_AREA:
             scale = np.sqrt(MIN_ADABINS_AREA) / np.sqrt(image_area)
             images = transforms.resize(
-                images, out_shape=(int(w * scale), int(h * scale)), resample="bicubic"
+                images, out_shape=(int(h * scale), int(w * scale)), resample="bicubic"
             )
 
         return self.model.predict(images)

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "perceptor"
-version = "0.6.5"
+version = "0.6.6"
 description = "Modular image generation library"
 authors = ["Richard Löwenström <samedii@gmail.com>", "dribnet"]
 readme = "README.md"