aboutsummaryrefslogtreecommitdiff
path: root/cgan.py
diff options
context:
space:
mode:
Diffstat (limited to 'cgan.py')
-rw-r--r--cgan.py6
1 files 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