diff options
Diffstat (limited to 'cgan.py')
| -rw-r--r-- | cgan.py | 6 |
1 files changed, 4 insertions, 2 deletions
@@ -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 |