diff options
author | Vasil Zlatanov <v@skozl.com> | 2019-03-13 20:03:09 +0000 |
---|---|---|
committer | Vasil Zlatanov <v@skozl.com> | 2019-03-13 20:03:09 +0000 |
commit | 03f2c41ac69084cde7a61eb04303078e3c4785a7 (patch) | |
tree | 5ccbe9a5f3714385a76a6b26546fc84398098680 /cdcgan.py | |
parent | 95de6b8e13302311ae2923818a8ac224b2c9fcc8 (diff) | |
download | e4-gan-03f2c41ac69084cde7a61eb04303078e3c4785a7.tar.gz e4-gan-03f2c41ac69084cde7a61eb04303078e3c4785a7.tar.bz2 e4-gan-03f2c41ac69084cde7a61eb04303078e3c4785a7.zip |
Update cdcgan with *better* one
Diffstat (limited to 'cdcgan.py')
-rwxr-xr-x | cdcgan.py | 20 |
1 files changed, 7 insertions, 13 deletions
@@ -65,22 +65,16 @@ class CDCGAN(): model.add(Dense(128 * 7 * 7, activation="relu", input_dim=self.latent_dim)) model.add(Reshape((7, 7, 128))) - model.add(UpSampling2D()) - for i in range(self.conv_layers): - model.add(Conv2D(128, kernel_size=3, padding="same")) - model.add(BatchNormalization()) - model.add(Activation("relu")) - - model.add(UpSampling2D()) - - for i in range(self.conv_layers): - model.add(Conv2D(64, kernel_size=3, padding="same")) - model.add(BatchNormalization()) + model.add(Conv2DTranspose(256, kernel_size=3, padding="same", strides=(2,2))) + model.add(BatchNormalization()) + model.add(Activation("relu")) - model.add(Activation("relu")) + model.add(Conv2DTranspose(128, kernel_size=3, padding="same", strides=(2,2))) + model.add(BatchNormalization()) + model.add(Activation("relu")) - model.add(Conv2D(self.channels, kernel_size=3, padding="same")) + model.add(Conv2DTranspose(64, kernel_size=3, padding="same")) model.add(Activation("tanh")) #model.summary() |