Update ip_adapter_inpainting.py

ip_adapter_inpainting.py

@@ -5,10 +5,10 @@ import torch
 
 from segmentation import get_cropped, get_blurred_mask, init_body as init_body_seg, init_face as init_face_seg
 from img2txt import derive_caption, init as init_img2txt
-from utils import alpha_composite_manuel
+from utils import alpha_composite
 from adapter_model import MODEL
 
-#init_face_seg()
+init_face_seg()
 init_body_seg()
 init_img2txt()
 
@@ -16,23 +16,20 @@ ip_model = MODEL("inpaint")
 
 
 def generate(img_openpose_gen: Image, img_human: Image, img_clothes: Image, segment_id: int):
-    cropped_clothes = get_cropped(img_openpose_gen, segment_id, False, False).resize(img_openpose_gen.size) 
-    cropped_body = get_cropped(img_human, segment_id, True, False).resize(img_openpose_gen.size) 
+    cropped_clothes = get_cropped(img_openpose_gen, segment_id, False, False).resize((512, 768)) 
+    cropped_body = get_cropped(img_human, segment_id, True, False).resize((512, 768))
 
     composite = Image.alpha_composite(cropped_body.convert('RGBA'),
                                       cropped_clothes.convert('RGBA')
                                 )
     composite = alpha_composite_manuel(composite)
 
-    input_clothes = get_cropped(img_clothes, segment_id, False, False).resize(img_openpose_gen.size)
-    input_clothes = alpha_composite_manuel(input_clothes)
-    
-    mask = get_blurred_mask(img_openpose_gen, segment_id)
+    mask = get_blurred_mask(composite, segment_id, False, False)
     prompt = derive_caption(img_clothes)
 
     ip_gen = ip_model.model.generate(
         prompt=prompt,
-        pil_image=input_clothes,
+        pil_image=img_clothes,
         num_samples=1,
         num_inference_steps=50,
         seed=42,
@@ -43,9 +40,9 @@ def generate(img_openpose_gen: Image, img_human: Image, img_clothes: Image, segm
         scale=0.8
         )[0]
 
-    #cropped_head = get_cropped(img_human.resize(img_openpose_gen.size), 13, False, True)
-    #ip_gen_final = Image.alpha_composite(ip_gen.convert("RGBA"),
-    #                                     cropped_head.convert("RGBA")
-    #                               )
+    cropped_head = get_cropped(composite, 13, False, True)
+    ip_gen_final = Image.alpha_composite(ip_gen.convert("RGBA"),
+                                         cropped_head.convert("RGBA")
+                                   )
     torch.cuda.empty_cache()
-    return ip_gen
+    return ip_gen_final