修复输出结果的比例错误的问题
Browse files
app.py
CHANGED
@@ -132,7 +132,10 @@ def infer(person,garment,denoise_steps,seed):
|
|
132 |
pipe.to(device)
|
133 |
pipe.unet_encoder.to(device)
|
134 |
|
135 |
-
|
|
|
|
|
|
|
136 |
garm_img= garment.convert("RGB").resize((768,1024))
|
137 |
|
138 |
progress(0.1,desc="Mask generating")
|
@@ -223,8 +226,9 @@ def infer(person,garment,denoise_steps,seed):
|
|
223 |
guidance_scale=2.0,
|
224 |
callback_on_step_end=callback
|
225 |
)[0]
|
|
|
226 |
progress(1,desc="Complete")
|
227 |
-
return
|
228 |
|
229 |
|
230 |
title = "## IDM-VTON"
|
|
|
132 |
pipe.to(device)
|
133 |
pipe.unet_encoder.to(device)
|
134 |
|
135 |
+
personRGB = person.convert("RGB")
|
136 |
+
crop_size = personRGB.size
|
137 |
+
|
138 |
+
human_img = personRGB.resize((768,1024))
|
139 |
garm_img= garment.convert("RGB").resize((768,1024))
|
140 |
|
141 |
progress(0.1,desc="Mask generating")
|
|
|
226 |
guidance_scale=2.0,
|
227 |
callback_on_step_end=callback
|
228 |
)[0]
|
229 |
+
out_img = images[0].resize(crop_size)
|
230 |
progress(1,desc="Complete")
|
231 |
+
return out_img
|
232 |
|
233 |
|
234 |
title = "## IDM-VTON"
|