Fixed padding error in ResizeMaxSize

mlfoundations · Sep 14, 2023 · f692ec9 · f692ec9
1 parent a4b5bc3
commit f692ec9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/open_clip/transform.py b/src/open_clip/transform.py
@@ -40,9 +40,10 @@ def forward(self, img):
         else:
             width, height = img.size
         scale = self.max_size / float(max(height, width))
+        new_size = tuple(round(dim * scale) for dim in (height, width))
         if scale != 1.0:
-            new_size = tuple(round(dim * scale) for dim in (height, width))
             img = F.resize(img, new_size, self.interpolation)
+        if not width == height:
             pad_h = self.max_size - new_size[0]
             pad_w = self.max_size - new_size[1]
             img = F.pad(img, padding=[pad_w//2, pad_h//2, pad_w - pad_w//2, pad_h - pad_h//2], fill=self.fill)