From 303232743c652af004c380992b53a89c1d7186f6 Mon Sep 17 00:00:00 2001
From: Jean-Luc Parouty <Jean-Luc.Parouty@simap.grenoble-inp.fr>
Date: Sat, 26 Feb 2022 10:37:18 +0100
Subject: [PATCH] Update GAN
---
DCGAN/01-DCGAN-Draw-me-a-sheep.ipynb | 2 +-
DCGAN/modules/models/DCGAN.py | 9 ++++++++-
2 files changed, 9 insertions(+), 2 deletions(-)
diff --git a/DCGAN/01-DCGAN-Draw-me-a-sheep.ipynb b/DCGAN/01-DCGAN-Draw-me-a-sheep.ipynb
index b6470ec..005dcad 100644
--- a/DCGAN/01-DCGAN-Draw-me-a-sheep.ipynb
+++ b/DCGAN/01-DCGAN-Draw-me-a-sheep.ipynb
@@ -401,7 +401,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
- "version": "3.9.7"
+ "version": "3.9.2"
}
},
"nbformat": 4,
diff --git a/DCGAN/modules/models/DCGAN.py b/DCGAN/modules/models/DCGAN.py
index 17a1bef..aa71490 100644
--- a/DCGAN/modules/models/DCGAN.py
+++ b/DCGAN/modules/models/DCGAN.py
@@ -63,6 +63,13 @@ class DCGAN(keras.Model):
discriminator_optimizer = keras.optimizers.Adam(),
generator_optimizer = keras.optimizers.Adam(),
loss_function = keras.losses.BinaryCrossentropy() ):
+ '''
+ Compile the model
+ args:
+ discriminator_optimizer : Discriminator optimizer (Adam)
+ generator_optimizer : Generator optimizer (Adam)
+ loss_function : Loss function
+ '''
super(DCGAN, self).compile()
self.d_optimizer = discriminator_optimizer
self.g_optimizer = generator_optimizer
@@ -147,7 +154,7 @@ class DCGAN(keras.Model):
# Sample random points in the latent space
random_latent_vectors = tf.random.normal(shape=(batch_size, self.latent_dim))
- # Assemble labels that say "all real images"
+ # Assemble labels that say all images are real, yes it's a lie ;-)
misleading_labels = tf.zeros((batch_size, 1))
# ---- Train the generator ---------------------------------
--
GitLab