Commit ff6acd35 authored by lenankamp's avatar lenankamp Committed by GitHub

Update img2img.py

Hopefully corrected the white space issue
parent bbce1673
...@@ -20,14 +20,14 @@ import modules.scripts ...@@ -20,14 +20,14 @@ import modules.scripts
def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args): def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args):
processing.fix_seed(p) processing.fix_seed(p)
# recursive batch, as written limits potential inputs to common image formats, may e better to just check if isfile for general use images = []
images = []
for root, directories, files in os.walk(input_dir): for root, directories, files in os.walk(input_dir):
for filename in files: for filename in files:
filepath = os.path.join(root, filename) filepath = os.path.join(root, filename)
if filepath.endswith(".jpg") or filepath.endswith(".jpeg") or filepath.endswith(".png") or filepath.endswith(".webp"): if filepath.endswith(".jpg") or filepath.endswith(".jpeg") or filepath.endswith(".png") or filepath.endswith(".webp"):
images.append(filepath) images.append(filepath)
is_inpaint_batch = False is_inpaint_batch = False
if inpaint_mask_dir: if inpaint_mask_dir:
inpaint_masks = shared.listfiles(inpaint_mask_dir) inpaint_masks = shared.listfiles(inpaint_mask_dir)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment