aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xcdcgan.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/cdcgan.py b/cdcgan.py
index f2122b8..effc89b 100755
--- a/cdcgan.py
+++ b/cdcgan.py
@@ -75,9 +75,12 @@ class CDCGAN():
model.add(Activation("relu"))
model.add(Conv2DTranspose(64, kernel_size=3, padding="same"))
+ model.add(BatchNormalization())
+ model.add(Activation("relu"))
+
+ model.add(Conv2DTranspose(1, kernel_size=3, padding="same"))
model.add(Activation("tanh"))
- #model.summary()
noise = Input(shape=(self.latent_dim,))
label = Input(shape=(1,), dtype='int32')
@@ -85,6 +88,8 @@ class CDCGAN():
model_input = multiply([noise, label_embedding])
img = model(model_input)
+ #model.summary()
+
return Model([noise, label], img)
def build_discriminator(self):