fix(api): resize blend mask to match source
This commit is contained in:
parent
1fbee0ae52
commit
2c2eda8c3c
|
@ -527,11 +527,15 @@ def run_blend_pipeline(
|
||||||
chain = ChainPipeline()
|
chain = ChainPipeline()
|
||||||
tile_size = get_base_tile(params, size)
|
tile_size = get_base_tile(params, size)
|
||||||
|
|
||||||
|
# resize mask to match source size
|
||||||
|
stage_source = sources[1]
|
||||||
|
stage_mask = mask.resize(stage_source.size, Image.Resampling.BILINEAR)
|
||||||
|
|
||||||
chain.stage(
|
chain.stage(
|
||||||
BlendMaskStage(),
|
BlendMaskStage(),
|
||||||
StageParams(tile_size=tile_size),
|
StageParams(tile_size=tile_size),
|
||||||
stage_source=sources[1],
|
stage_source=stage_source,
|
||||||
stage_mask=mask,
|
stage_mask=stage_mask,
|
||||||
)
|
)
|
||||||
|
|
||||||
# apply upscaling and correction
|
# apply upscaling and correction
|
||||||
|
|
Loading…
Reference in New Issue