Skip to content

Commit 525c1ed

Browse files
author
k_sugawara
committed
make save dir if save dir is not exists
1 parent 5c9b362 commit 525c1ed

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

modules/img2img.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ def process_batch(p, input_dir, output_dir, args):
5555
filename = f"{left}-{n}{right}"
5656

5757
if not save_normally:
58+
os.makedirs(output_dir, exist_ok=True)
5859
processed_image.save(os.path.join(output_dir, filename))
5960

6061

0 commit comments

Comments
 (0)