diff options
| author | missionfloyd <missionfloyd@users.noreply.github.com> | 2023-03-27 03:47:05 +0000 |
|---|---|---|
| committer | missionfloyd <missionfloyd@users.noreply.github.com> | 2023-03-27 03:47:05 +0000 |
| commit | efac2cf1ab6645f3f5134158c1401c6305c2ffea (patch) | |
| tree | 24b69e980c07d03618a6e5b2447704cdb30a6a20 /modules/processing.py | |
| parent | 1d096ed1456c9b9b662477839853621848705e68 (diff) | |
| parent | a336c7fe233fa7dff062f5187c0f4d01ab26e80b (diff) | |
| download | stable-diffusion-webui-gfx803-efac2cf1ab6645f3f5134158c1401c6305c2ffea.tar.gz stable-diffusion-webui-gfx803-efac2cf1ab6645f3f5134158c1401c6305c2ffea.tar.bz2 stable-diffusion-webui-gfx803-efac2cf1ab6645f3f5134158c1401c6305c2ffea.zip | |
Merge branch 'extra-network-preview-lazyload' of https://github.com/missionfloyd/stable-diffusion-webui into extra-network-preview-lazyload
Diffstat (limited to 'modules/processing.py')
| -rw-r--r-- | modules/processing.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py index 59717b4c..2e5a363f 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -689,6 +689,22 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: image.info["parameters"] = text
output_images.append(image)
+ if hasattr(p, 'mask_for_overlay') and p.mask_for_overlay:
+ image_mask = p.mask_for_overlay.convert('RGB')
+ image_mask_composite = Image.composite(image.convert('RGBA').convert('RGBa'), Image.new('RGBa', image.size), p.mask_for_overlay.convert('L')).convert('RGBA')
+
+ if opts.save_mask:
+ images.save_image(image_mask, p.outpath_samples, "", seeds[i], prompts[i], opts.samples_format, info=infotext(n, i), p=p, suffix="-mask")
+
+ if opts.save_mask_composite:
+ images.save_image(image_mask_composite, p.outpath_samples, "", seeds[i], prompts[i], opts.samples_format, info=infotext(n, i), p=p, suffix="-mask-composite")
+
+ if opts.return_mask:
+ output_images.append(image_mask)
+
+ if opts.return_mask_composite:
+ output_images.append(image_mask_composite)
+
del x_samples_ddim
devices.torch_gc()
|