StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ruffed

Browse Source
This commit is contained in:
Artem Kotov 2023-05-29 21:38:49 +04:00
parent 6c610a8a95
commit 23314a6e27
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/img2img.py
View File

@ -50,11 +50,11 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
continue continue
# Use the EXIF orientation of photos taken by smartphones. # Use the EXIF orientation of photos taken by smartphones.
img = ImageOps.exif_transpose(img) img = ImageOps.exif_transpose(img)
if to_scale: if to_scale:
p.width = int(img.width * scale_by) p.width = int(img.width * scale_by)
p.height = int(img.height * scale_by) p.height = int(img.height * scale_by)
p.init_images = [img] * p.batch_size p.init_images = [img] * p.batch_size
if is_inpaint_batch: if is_inpaint_batch:
@ -63,7 +63,7 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
mask_image_path = list(path.parent.glob(f"**/{path.stem}*")) mask_image_path = list(path.parent.glob(f"**/{path.stem}*"))
if len(mask_image_path) > 0: if len(mask_image_path) > 0:
mask_image_path = str(mask_image_path[0]) mask_image_path = str(mask_image_path[0])
# if not found use first one ("same mask for all images" use-case) # if not found use first one ("same mask for all images" use-case)
if mask_image_path not in inpaint_masks: if mask_image_path not in inpaint_masks:
mask_image_path = inpaint_masks[0] mask_image_path = inpaint_masks[0]