altayavci commited on
Commit
94228f2
·
1 Parent(s): 564fba9

Update ip_adapter_inpainting.py

Browse files
Files changed (1) hide show
  1. 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(cropped_body.convert('RGBA'),
23
- cropped_clothes.convert('RGBA')
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(ip_gen.convert("RGBA"),
45
- cropped_head.convert("RGBA")
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