From 03f2c41ac69084cde7a61eb04303078e3c4785a7 Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Wed, 13 Mar 2019 20:03:09 +0000 Subject: Update cdcgan with *better* one --- dcgan.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'dcgan.py') diff --git a/dcgan.py b/dcgan.py index 347f61e..719e096 100644 --- a/dcgan.py +++ b/dcgan.py @@ -1,4 +1,5 @@ from __future__ import print_function, division +import tensorflow.keras as keras from keras.datasets import mnist from keras.layers import Input, Dense, Reshape, Flatten, Dropout from keras.layers import BatchNormalization, Activation, ZeroPadding2D @@ -75,7 +76,7 @@ class DCGAN(): model.add(Conv2D(self.channels, kernel_size=3, padding="same")) model.add(Activation("tanh")) - #model.summary() + model.summary() noise = Input(shape=(self.latent_dim,)) img = model(noise) @@ -187,8 +188,6 @@ class DCGAN(): fig.savefig("images/mnist_%d.png" % epoch) plt.close() -''' if __name__ == '__main__': dcgan = DCGAN() dcgan.train(epochs=4000, batch_size=32, save_interval=50) -''' -- cgit v1.2.3-54-g00ecf From 9b71f4e88cab0f1981dc0e2a846510894756ea8a Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Wed, 13 Mar 2019 20:05:24 +0000 Subject: Add import --- dcgan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'dcgan.py') diff --git a/dcgan.py b/dcgan.py index a362f69..f65f742 100644 --- a/dcgan.py +++ b/dcgan.py @@ -4,7 +4,7 @@ from keras.datasets import mnist from keras.layers import Input, Dense, Reshape, Flatten, Dropout from keras.layers import BatchNormalization, Activation, ZeroPadding2D from keras.layers.advanced_activations import LeakyReLU -from keras.layers.convolutional import UpSampling2D, Conv2D +from keras.layers.convolutional import UpSampling2D, Conv2D, Conv2DTranspose from keras.models import Sequential, Model from keras.optimizers import Adam -- cgit v1.2.3-54-g00ecf From a5f79e9ad499681c77630f71d00c95715b5f6b9e Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Wed, 13 Mar 2019 20:05:48 +0000 Subject: Revert "Add import" This reverts commit 9b71f4e88cab0f1981dc0e2a846510894756ea8a. --- dcgan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'dcgan.py') diff --git a/dcgan.py b/dcgan.py index f65f742..a362f69 100644 --- a/dcgan.py +++ b/dcgan.py @@ -4,7 +4,7 @@ from keras.datasets import mnist from keras.layers import Input, Dense, Reshape, Flatten, Dropout from keras.layers import BatchNormalization, Activation, ZeroPadding2D from keras.layers.advanced_activations import LeakyReLU -from keras.layers.convolutional import UpSampling2D, Conv2D, Conv2DTranspose +from keras.layers.convolutional import UpSampling2D, Conv2D from keras.models import Sequential, Model from keras.optimizers import Adam -- cgit v1.2.3-54-g00ecf