Add prefix to output images

Psycho7 · Oct 25, 2017 · a163f94 · a163f94
1 parent 9778c7d
commit a163f94
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/autoencoder.py b/autoencoder.py
@@ -434,7 +434,7 @@ def get_weights_as_images(self, width, height, outdir='img/', max_images=10, mod
                 image_path = outdir + self.model_name + '-enc_weights_{}.png'.format(p)
                 utils.gen_image(enc_w, width, height, image_path)
 
-    def get_images(self, images, channel=3, width=32, height=32,  outdir='img/', model_path=None):
+    def get_images(self, images, prefix, channel=3, width=32, height=32,  outdir='img/', model_path=None):
         assert channel in [1, 3]
 
         outdir = self.data_dir + outdir
@@ -461,17 +461,17 @@ def get_images(self, images, channel=3, width=32, height=32,  outdir='img/', mod
                     origin,
                     width,
                     height,
-                    '%s%s-origin-%d.png' % (outdir, self.model_name, i),
+                    '%s%s-%s-origin-%d.png' % (outdir, self.model_name, prefix, i),
                     image_type)
                 utils.gen_image(
                     corrupted,
                     width,
                     height,
-                    '%s%s-corrupted-%d.png' % (outdir, self.model_name, i),
+                    '%s%s-%s-corrupted-%d.png' % (outdir, self.model_name, prefix, i),
                     image_type)
                 utils.gen_image(
                     decoded,
                     width,
                     height,
-                    '%s%s-decoded-%d.png' % (outdir, self.model_name, i),
+                    '%s%s-%s-decoded-%d.png' % (outdir, self.model_name, prefix, i),
                     image_type)
diff --git a/run_autoencoder.py b/run_autoencoder.py
@@ -90,7 +90,7 @@
     # dae.get_weights_as_images(28, 28, max_images=FLAGS.weight_images)
     index = [1100, 2200, 3300, 4400, 5500, 6600, 7700]
     trImg, teImg = [trX[index], teX[index]]
-    dae.get_images(trImg)
-    dae.get_images(teImg)
+    dae.get_images(trImg, "train")
+    dae.get_images(teImg, "test")