diff --git a/api/onnx_web/image.py b/api/onnx_web/image.py index 6093b3f8..de0d3f43 100644 --- a/api/onnx_web/image.py +++ b/api/onnx_web/image.py @@ -179,7 +179,8 @@ def expand_image( full_mask = mask_filter(mask_image, dims, origin) full_noise = noise_source(source_image, dims, origin) - # TODO: multiply noise by mask before compositing + full_noise = ImageChops.multiply(full_noise, full_mask) + full_source = Image.composite(full_noise, full_source, full_mask.convert('L')) return (full_source, full_mask, full_noise, (full_width, full_height))