keep reg_w and reg_h unchange if it's already multiply of strides

Pandede · Apr 10, 2024 · 359cd1e · 359cd1e
1 parent 39dcb0c
commit 359cd1e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/wpodnet/backend.py b/wpodnet/backend.py
@@ -62,8 +62,13 @@ def _resize_to_fixed_ratio(self, image: Image.Image, dim_min: int, dim_max: int)
         reg_w, reg_h = int(w * factor), int(h * factor)
 
         # Ensure the both width and height are the multiply of `self._stride`
-        reg_w += self._stride - reg_w % self._stride
-        reg_h += self._stride - reg_h % self._stride
+        reg_w_mod = reg_w % self._stride
+        if reg_w_mod > 0:
+            reg_w += self._stride - reg_w_mod
+
+        reg_h_mod = reg_h % self._stride
+        if reg_h_mod > 0:
+            reg_h += self._stride - reg_h % self._stride
 
         return image.resize((reg_w, reg_h))