From ccb22bee228213a92985bfb3d718eac959d852b0 Mon Sep 17 00:00:00 2001
From: kopytjuk <kopytjuk@gmail.com>
Date: Wed, 26 Dec 2018 16:03:28 +0100
Subject: [PATCH] Save each image in its own file.

---
 generate.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/generate.py b/generate.py
index 149fa9a..0c93261 100644
--- a/generate.py
+++ b/generate.py
@@ -56,7 +56,8 @@ if __name__=="__main__":
         netG.cuda()
         fixed_noise = fixed_noise.cuda()
 
-    fake = netG(Variable(fixed_noise, volatile=True))
+    fake = netG(fixed_noise)
     fake.data = fake.data.mul(0.5).add(0.5)
 
-    vutils.save_image(fake.data, os.path.join(opt.output_dir, "generated.png"))
+    for i in range(opt.nimages):
+        vutils.save_image(fake.data[i, ...].reshape((1, nc, imageSize, imageSize)), os.path.join(opt.output_dir, "generated_%02d.png"%i))