From 69eb6678fc1c61fdb77f84f25f29ba9f777cd3dd Mon Sep 17 00:00:00 2001 From: Vee9ahd1 Date: Mon, 3 Jun 2019 12:17:29 -0400 Subject: output directory and file prefix (base name) are now handled by different arguments --- gantools/image_utils.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'gantools/image_utils.py') diff --git a/gantools/image_utils.py b/gantools/image_utils.py index 016891f..72130b0 100644 --- a/gantools/image_utils.py +++ b/gantools/image_utils.py @@ -1,10 +1,11 @@ +import os import PIL.Image def save_image(arr, fp, format='JPEG'): image = PIL.Image.fromarray(arr) image.save(fp, format=format, quality=90) -def save_images(ims, path_prefix='', format='JPEG'): +def save_images(ims, output_dir='', prefix='', format='JPEG'): for i, im in enumerate(ims): - path = str(path_prefix)+str(i).zfill(4)+'.'+str(format).lower() - save_image(im, path) + full_path = os.path.join(output_dir, prefix + str(i).zfill(4) + '.' + format.lower()) + save_image(im, full_path, format) -- cgit v1.2.1