Keshabwi66 commited on
Commit
e992f1f
·
verified ·
1 Parent(s): 25d18e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -134,13 +134,15 @@ def start_tryon(person_img,cloth_img, garment_des, denoise_steps=10, seed=42):
134
  # Resize and prepare images
135
  garm_img = cloth_img.convert("RGB").resize((768, 1024))
136
  human_img = person_img.convert("RGB").resize((768, 1024))
 
137
  is_checked=True;
138
  if is_checked:
139
  keypoints = openpose_model(human_img.resize((384,512)))
140
  model_parse, _ = parsing_model(human_img.resize((384,512)))
141
  mask= get_mask_location('hd', "upper_body", model_parse, keypoints)
 
142
 
143
- mask = pil_to_binary_mask(mask.convert("RGB").resize((768, 1024)))
144
  pose_img=Image.open("00006_00.jpg")
145
 
146
  # Prepare pose image (already uploaded)
 
134
  # Resize and prepare images
135
  garm_img = cloth_img.convert("RGB").resize((768, 1024))
136
  human_img = person_img.convert("RGB").resize((768, 1024))
137
+
138
  is_checked=True;
139
  if is_checked:
140
  keypoints = openpose_model(human_img.resize((384,512)))
141
  model_parse, _ = parsing_model(human_img.resize((384,512)))
142
  mask= get_mask_location('hd', "upper_body", model_parse, keypoints)
143
+ mask = mask.resize((768,1024))
144
 
145
+ #mask = pil_to_binary_mask(mask.convert("RGB").resize((768, 1024)))
146
  pose_img=Image.open("00006_00.jpg")
147
 
148
  # Prepare pose image (already uploaded)