Update app.py

app.py

@@ -175,7 +175,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
     pipe.to(device)
     pipe.unet_encoder.to(device)
 
-    garm_img = garm_img.convert("RGB").resize((384, 512))
+    garm_img = garm_img.convert("RGB").resize((768, 1024))
     human_img_orig = dict["background"].convert("RGB")
 
     if is_checked_crop:
@@ -188,9 +188,9 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
         bottom = (height + target_height) / 2
         cropped_img = human_img_orig.crop((left, top, right, bottom))
         crop_size = cropped_img.size
-        human_img = cropped_img.resize((384, 512))
+        human_img = cropped_img.resize((768, 1024))
     else:
-        human_img = human_img_orig.resize((384, 512))
+        human_img = human_img_orig.resize((768, 1024))
 
     if is_checked:
         keypoints = openpose_model(human_img.resize((384, 512)))
@@ -198,7 +198,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
         mask, mask_gray = get_mask_location('hd', categorie , model_parse, keypoints)
         mask = mask.resize((768, 1024))
     else:
-        mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((384, 512)))
+        mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))
     mask_gray = (1 - transforms.ToTensor()(mask)) * tensor_transfrom(human_img)
     mask_gray = to_pil_image((mask_gray + 1.0) / 2.0)
 
@@ -208,7 +208,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
     args = apply_net.create_argument_parser().parse_args(('show', './configs/densepose_rcnn_R_50_FPN_s1x.yaml', './ckpt/densepose/model_final_162be9.pkl', 'dp_segm', '-v', '--opts', 'MODEL.DEVICE', 'cuda'))
     pose_img = args.func(args, human_img_arg)
     pose_img = pose_img[:, :, ::-1]
-    pose_img = Image.fromarray(pose_img).resize((384, 512))
+    pose_img = Image.fromarray(pose_img).resize((768, 1024))
 
     with torch.no_grad():
         with torch.cuda.amp.autocast():
@@ -264,7 +264,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
                     image=human_img,
                     height=1024,
                     width=768,
-                    ip_adapter_image=garm_img.resize((384, 512)),
+                    ip_adapter_image=garm_img.resize((768, 1024)),
                     guidance_scale=2.0,
                 )[0]