apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

always use alpha in blend stage

This commit is contained in:
Sean Sube 2023-12-14 20:12:39 -06:00
parent 95d8f4a598
commit 6c4f4f334f
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -30,17 +30,16 @@ class BlendMaskStage(BaseStage):
) -> StageResult: ) -> StageResult:
logger.info("blending image using mask") logger.info("blending image using mask")
# TODO: does this need an alpha channel?
mult_mask = Image.new(stage_mask.mode, stage_mask.size, color="black") mult_mask = Image.new(stage_mask.mode, stage_mask.size, color="black")
mult_mask.alpha_composite(stage_mask) mult_mask = Image.alpha_composite(mult_mask, stage_mask)
mult_mask = mult_mask.convert("L") mult_mask = mult_mask.convert("L")
if is_debug(): if is_debug():
save_image(server, "last-mask.png", stage_mask) save_image(server, "last-mask.png", stage_mask)
save_image(server, "last-mult-mask.png", mult_mask) save_image(server, "last-mult-mask.png", mult_mask)
return StageResult( return StageResult.from_images(
images=[ [
Image.composite(stage_source, source, mult_mask) Image.composite(stage_source, source, mult_mask)
for source in sources.as_image() for source in sources.as_image()
] ]