Spaces:
Configuration error
Configuration error
Update ip_adapter_inpainting.py
Browse files- ip_adapter_inpainting.py +4 -4
ip_adapter_inpainting.py
CHANGED
@@ -19,8 +19,8 @@ def generate(img_openpose_gen: Image, img_human: Image, img_clothes: Image, segm
|
|
19 |
cropped_clothes = get_cropped(img_openpose_gen, segment_id, False).resize((512, 768))
|
20 |
cropped_body = get_cropped(img_human, segment_id, True).resize((512, 768))
|
21 |
|
22 |
-
composite = alpha_composite(
|
23 |
-
|
24 |
)
|
25 |
composite = alpha_composite(composite)
|
26 |
|
@@ -41,8 +41,8 @@ def generate(img_openpose_gen: Image, img_human: Image, img_clothes: Image, segm
|
|
41 |
)[0]
|
42 |
|
43 |
cropped_head = get_cropped(composite, 13, False, True)
|
44 |
-
ip_gen_final = alpha_composite(
|
45 |
-
|
46 |
)
|
47 |
torch.cuda.empty_cache()
|
48 |
return ip_gen_final
|
|
|
19 |
cropped_clothes = get_cropped(img_openpose_gen, segment_id, False).resize((512, 768))
|
20 |
cropped_body = get_cropped(img_human, segment_id, True).resize((512, 768))
|
21 |
|
22 |
+
composite = alpha_composite(cropped_clothes.convert('RGBA'),
|
23 |
+
cropped_body.convert('RGBA')
|
24 |
)
|
25 |
composite = alpha_composite(composite)
|
26 |
|
|
|
41 |
)[0]
|
42 |
|
43 |
cropped_head = get_cropped(composite, 13, False, True)
|
44 |
+
ip_gen_final = alpha_composite(cropped_head.convert("RGBA"),
|
45 |
+
ip_gen.convert("RGBA")
|
46 |
)
|
47 |
torch.cuda.empty_cache()
|
48 |
return ip_gen_final
|