apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): handle blending images of different sizes better

This commit is contained in:
Sean Sube 2023-12-20 22:58:57 -06:00
parent e933dc6d50
commit de28ecdfad
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,5 +1,5 @@
from logging import getLogger
from typing import Optional
from typing import Optional, Tuple
from PIL import Image
@ -23,6 +23,7 @@ class BlendMaskStage(BaseStage):
_params: ImageParams,
sources: StageResult,
*,
dims: Tuple[int, int, int],
stage_source: Optional[Image.Image] = None,
stage_mask: Optional[Image.Image] = None,
_callback: Optional[ProgressCallback] = None,
@ -34,13 +35,17 @@ class BlendMaskStage(BaseStage):
mult_mask = Image.alpha_composite(mult_mask, stage_mask)
mult_mask = mult_mask.convert("L")
top, left, tile = dims
stage_source_tile = stage_source.crop((top, left, tile, tile))
if is_debug():
save_image(server, "last-mask.png", stage_mask)
save_image(server, "last-mult-mask.png", mult_mask)
save_image(server, "last-stage-source.png", stage_source_tile)
return StageResult.from_images(
[
Image.composite(stage_source, source, mult_mask)
Image.composite(stage_source_tile, source, mult_mask)
for source in sources.as_image()
]
)