Update app.py
Browse files
app.py
CHANGED
@@ -149,7 +149,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
|
|
149 |
if is_checked:
|
150 |
keypoints = openpose_model(human_img.resize((384, 512)))
|
151 |
model_parse, _ = parsing_model(human_img.resize((384, 512)))
|
152 |
-
mask, mask_gray = get_mask_location('hd', "
|
153 |
mask = mask.resize((768, 1024))
|
154 |
else:
|
155 |
mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))
|
|
|
149 |
if is_checked:
|
150 |
keypoints = openpose_model(human_img.resize((384, 512)))
|
151 |
model_parse, _ = parsing_model(human_img.resize((384, 512)))
|
152 |
+
mask, mask_gray = get_mask_location('hd', "full_body", model_parse, keypoints)
|
153 |
mask = mask.resize((768, 1024))
|
154 |
else:
|
155 |
mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))
|