apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): convert blend sources to the same size as the mask

This commit is contained in:
Sean Sube 2023-02-13 21:57:47 -06:00
parent 3544e238cf
commit d1b250660e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
api/onnx_web/chain/blend_mask.py
View File

@ -25,12 +25,15 @@ def blend_mask(
) -> Image.Image: ) -> Image.Image:
logger.info("blending image using mask") logger.info("blending image using mask")
l_mask = Image.new("RGBA", mask.size, color="black") mult_mask = Image.new("RGBA", mask.size, color="black")
l_mask.alpha_composite(mask) mult_mask.alpha_composite(mask)
l_mask = l_mask.convert("L") mult_mask = mult_mask.convert("L")
if is_debug(): if is_debug():
save_image(server, "last-mask.png", mask) save_image(server, "last-mask.png", mask)
save_image(server, "last-mask-l.png", l_mask) save_image(server, "last-mult-mask.png", mult_mask)
return Image.composite(sources[0], sources[1], l_mask) for source in sources:
source.thumbnail(mult_mask.size)
return Image.composite(sources[0], sources[1], mult_mask)