2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed dimensions of input transform

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-18 12:17:01 +02:00
parent 71623db905
commit 2140b13857
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/aae/model.py
View File

@ -90,7 +90,7 @@ class Decoder(keras.Model):
weight_init = keras.initializers.RandomNormal(mean=0, stddev=0.02)
# calculate dimension of last conv layer in encoder
conv_image_size = image_size / (2 ** 3)
dimensions = zsize * conv_image_size
dimensions = zsize * conv_image_size * conv_image_size
self.conv_shape = (-1, conv_image_size, conv_image_size, zsize)
self.transform = keras.layers.Dense(units=dimensions, name='input_transform')
self.deconv1 = keras.layers.Conv2DTranspose(filters=zsize, kernel_size=3, strides=1, name='deconv1',