Fix image rescaling to [0,1]

kavehahmadi60 · Feb 15, 2019 · 1a0428a · 1a0428a
1 parent 77ed721
commit 1a0428a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/sgan/sgan.py b/sgan/sgan.py
@@ -175,7 +175,7 @@ def sample_images(self, epoch):
         gen_imgs = self.generator.predict(noise)
 
         # Rescale images 0 - 1
-        gen_imgs = 0.5 * gen_imgs + 1
+        gen_imgs = 0.5 * gen_imgs + 0.5
 
         fig, axs = plt.subplots(r, c)
         cnt = 0

diff --git a/wgan/wgan.py b/wgan/wgan.py
@@ -173,7 +173,7 @@ def sample_images(self, epoch):
         gen_imgs = self.generator.predict(noise)
 
         # Rescale images 0 - 1
-        gen_imgs = 0.5 * gen_imgs + 1
+        gen_imgs = 0.5 * gen_imgs + 0.5
 
         fig, axs = plt.subplots(r, c)
         cnt = 0

diff --git a/wgan_gp/wgan_gp.py b/wgan_gp/wgan_gp.py
@@ -227,7 +227,7 @@ def sample_images(self, epoch):
         gen_imgs = self.generator.predict(noise)
 
         # Rescale images 0 - 1
-        gen_imgs = 0.5 * gen_imgs + 1
+        gen_imgs = 0.5 * gen_imgs + 0.5
 
         fig, axs = plt.subplots(r, c)
         cnt = 0