From 723ac9bb7094b482a2efc6bc97139409d2bde9c4 Mon Sep 17 00:00:00 2001 From: nunzip Date: Wed, 27 Feb 2019 23:10:04 +0000 Subject: Save images for all classes instead of zero only --- cgan.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/cgan.py b/cgan.py index 3b3d0b7..716e5ae 100644 --- a/cgan.py +++ b/cgan.py @@ -175,9 +175,11 @@ class CGAN(): r, c = 2, 5 noise = np.random.normal(0, 1, (r * c, 100)) sampled_labels = np.arange(0, 10).reshape(-1, 1) - dummy_labels = np.zeros(32).reshape(-1, 1) + + #using dummy_labels would just print zeros to help identify image quality + #dummy_labels = np.zeros(32).reshape(-1, 1) - gen_imgs = self.generator.predict([noise, dummy_labels]) + gen_imgs = self.generator.predict([noise, sampled_labels]) # Rescale images 0 - 1 gen_imgs = 0.5 * gen_imgs + 0.5 -- cgit v1.2.3-54-g00ecf