diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-29 05:11:03 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-29 05:11:03 +0000 |
| commit | a1e5e0d7669def010ecf31d801d6f0667bcf8061 (patch) | |
| tree | 640f6a5abb607c9553731502dbcd4497c5677d11 | |
| parent | cf8da8e1b0b8a4ea75c66ed4657f409cadab2c59 (diff) | |
| download | stable-diffusion-webui-gfx803-a1e5e0d7669def010ecf31d801d6f0667bcf8061.tar.gz stable-diffusion-webui-gfx803-a1e5e0d7669def010ecf31d801d6f0667bcf8061.tar.bz2 stable-diffusion-webui-gfx803-a1e5e0d7669def010ecf31d801d6f0667bcf8061.zip | |
skip filenames starting with . for img2img and extras batch modes
| -rw-r--r-- | modules/extras.py | 2 | ||||
| -rw-r--r-- | modules/img2img.py | 2 | ||||
| -rw-r--r-- | modules/shared.py | 5 |
3 files changed, 7 insertions, 2 deletions
diff --git a/modules/extras.py b/modules/extras.py index 681d8d5a..4d51088b 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -72,7 +72,7 @@ def run_extras(extras_mode, resize_mode, image, image_folder, input_dir, output_ if input_dir == '':
return outputs, "Please select an input directory.", ''
- image_list = [file for file in [os.path.join(input_dir, x) for x in sorted(os.listdir(input_dir))] if os.path.isfile(file)]
+ image_list = shared.listfiles(input_dir)
for img in image_list:
try:
image = Image.open(img)
diff --git a/modules/img2img.py b/modules/img2img.py index 9c0cf23e..efda26e1 100644 --- a/modules/img2img.py +++ b/modules/img2img.py @@ -19,7 +19,7 @@ import modules.scripts def process_batch(p, input_dir, output_dir, args):
processing.fix_seed(p)
- images = [file for file in [os.path.join(input_dir, x) for x in os.listdir(input_dir)] if os.path.isfile(file)]
+ images = shared.listfiles(input_dir)
print(f"Will process {len(images)} images, creating {p.n_iter * p.batch_size} new images for each.")
diff --git a/modules/shared.py b/modules/shared.py index 7c428d90..7e634423 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -450,3 +450,8 @@ total_tqdm = TotalTQDM() mem_mon = modules.memmon.MemUsageMonitor("MemMon", device, opts)
mem_mon.start()
+
+
+def listfiles(dirname):
+ filenames = [os.path.join(dirname, x) for x in sorted(os.listdir(dirname)) if not x.startswith(".")]
+ return [file for file in filenames if os.path.isfile(file)]
|