From 4357ff77007f7aaaa1dc40654896820e969ac085 Mon Sep 17 00:00:00 2001
From: Great <44393369+yaosheng216@users.noreply.github.com>
Date: Mon, 21 Apr 2025 14:33:43 +0800
Subject: [PATCH] Update ace_plus_dataset.py

---
 modules/ace_plus_dataset.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/modules/ace_plus_dataset.py b/modules/ace_plus_dataset.py
index 2033f62..88305ff 100644
--- a/modules/ace_plus_dataset.py
+++ b/modules/ace_plus_dataset.py
@@ -26,15 +26,14 @@ def transform_image(image, std = 0.5, mean = 0.5):
     return (image.permute(2, 0, 1)/255. - mean)/std
 def transform_mask(mask):
     return mask.unsqueeze(0)/255.
+    
 def ensure_src_align_target_h_mode(src_image, size, image_id, interpolation=InterpolationMode.BILINEAR):
     # padding mode
     H, W = size
     ret_image = []
     for one_id in image_id:
         edit_image = src_image[one_id]
-        _, eH, eW = edit_image.shape
-        scale = H/eH
-        tH, tW = H, int(eW * scale)
+        tH, tW = H, W
         ret_image.append(T.Resize((tH, tW), interpolation=interpolation, antialias=True)(edit_image))
     return ret_image