22

utils/image_utils.py

@@ -11,7 +11,7 @@ def rescale_image(img, scale, nearest=32, max_size=1280):
         # Calculate new size keeping aspect ratio
         scale = min(max_size / new_w, max_size / new_h)
         new_w = int(new_w * scale)
-        new_w = int(new_w * scale)
+        new_h = int(new_h * scale)
 
     # Adjust to nearest multiple
     new_w = (new_w // nearest) * nearest