图片调整回原始尺寸

app.py

@@ -147,6 +147,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
     # 2. 图像预处理 - 调整服装和人物图像大小
     garm_img = garm_img.convert("RGB").resize((768,1024))
     human_img_orig = dict["background"].convert("RGB")
+    orig_size = human_img_orig.size  # 保存原始尺寸
 
     # 2.1 如果启用裁剪，按3:4比例裁剪人物图像
     if is_checked_crop:
@@ -260,12 +261,11 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
 
     # 7. 后处理 - 处理裁剪情况并返回结果
     if is_checked_crop:
-        out_img = images[0].resize(crop_size)
-        human_img_orig.paste(out_img, (int(left), int(top)))
-        return human_img_orig, mask_gray
+        # 将生成的图片和mask调整到裁剪后的尺寸
+        return images[0].resize(crop_size), mask_gray.resize(crop_size)
     else:
-        return images[0], mask_gray
-    # return images[0], mask_gray
+        # 直接将生成的图片和mask调整到原始尺寸
+        return images[0].resize(orig_size), mask_gray.resize(orig_size)
 
 garm_list = os.listdir(os.path.join(example_path,"cloth"))
 garm_list_path = [os.path.join(example_path,"cloth",garm) for garm in garm_list]
@@ -344,11 +344,9 @@ with image_blocks as demo:
         gr.Markdown("###### [Change Clothes AI](https://changeclothesai.online/)")
         gr.Markdown("###### [Describe Image AI](https://describeimageai.online/)")
 
-
     try_button.click(fn=start_tryon, inputs=[imgs, garm_img, prompt, is_checked,is_checked_crop, denoise_steps, seed, category], outputs=[image_out,masked_img], api_name='tryon')
 
 
 
 
 image_blocks.launch()
-