change resize

pipeline.py

@@ -6060,15 +6060,15 @@ class MaxLongEdgeMinShortEdgeResize(torch.nn.Module):
             scale = self.max_size / max(new_width, new_height)
             new_width, new_height = self._apply_scale(new_width, new_height, scale)
         
-        if img.dim() == 3:
-            img = img.unsqueeze(0)  # → [1,C,H,W]
+        # if img.dim() == 3:
+        #     img = img.unsqueeze(0)  # → [1,C,H,W]
         resized = F.interpolate(
             img,
-            size=(new_h, new_w),
-            mode=interpolation,     # e.g. "bilinear"
-            antialias=True,         # if you need anti‑aliasing
+            size=(new_height, new_width),
+            mode=self.interpolation,     # e.g. "bilinear"
+            antialias=self.antialias,         # if you need anti‑aliasing
         )
-        resized = resized.squeeze(0)  # → [C,H,W]
+        # resized = resized.squeeze(0)  # → [C,H,W]
         return resized
         # return F.resize(img, (new_height, new_width), self.interpolation, antialias=self.antialias)