michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

convert rgba to rgb

Browse source 
some image format (e.g. jpg) do not support rgba
This commit is contained in:
Vladimir Mandic 2023-02-08 07:03:36 -05:00 committed by GitHub
parent ea9bd9fc74
commit 4c562a9832
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/img2img.py
View file

@ -73,6 +73,8 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args):
if not save_normally: if not save_normally:
os.makedirs(output_dir, exist_ok=True) os.makedirs(output_dir, exist_ok=True)
if processed_image.mode == 'RGBA':
processed_image = processed_image.convert("RGB")
processed_image.save(os.path.join(output_dir, filename)) processed_image.save(os.path.join(output_dir, filename))