Merge pull request #4087 from ikasumi/feat/batch-img2img-improve
make save dir if save dir is not exists
This commit is contained in:
commit
c28de154b0
1 changed files with 1 additions and 0 deletions
|
@ -55,6 +55,7 @@ def process_batch(p, input_dir, output_dir, args):
|
||||||
filename = f"{left}-{n}{right}"
|
filename = f"{left}-{n}{right}"
|
||||||
|
|
||||||
if not save_normally:
|
if not save_normally:
|
||||||
|
os.makedirs(output_dir, exist_ok=True)
|
||||||
processed_image.save(os.path.join(output_dir, filename))
|
processed_image.save(os.path.join(output_dir, filename))
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue