1 files changed, 10 insertions, 0 deletions
diff --git a/gantools/image_utils.py b/gantools/image_utils.py
new file mode 100644
index 0000000..ca67134
--- /dev/null
+++ b/gantools/image_utils.py
@@ -0,0 +1,10 @@
+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'):
+    for i, im in enumerate(ims):
+        path = str(path_prefix)+str(i).zfill(4)+'.'+str(format).lower()
+        save_image(im, path)