Вход 0 несовместим со слоем conv2d_transpose_1: ожидаемый ndim = 4, найденный ndim = 2 - PullRequest
Новая
       45

Вход 0 несовместим со слоем conv2d_transpose_1: ожидаемый ndim = 4, найденный ndim = 2

0 голосов
/ 26 июня 2018

У меня проблемы с изменением формы слоя перед подачей на него в результате деконволюции. Я не знаю, как перевернуть плоский слой в свертке. Спасибо за помощь! 

def build_deep_autoencoder(img_shape, code_size):
H,W,C = img_shape
encoder = keras.models.Sequential()
encoder.add(L.InputLayer(img_shape))
encoder.add(L.Conv2D(32, (3,3), padding = 'same', activation = 'elu', name='layer_1'))
encoder.add(L.MaxPooling2D((3,3), padding = 'same',name = 'max_pooling_1'))
encoder.add(L.Conv2D(64, (3,3), padding = 'same', activation = 'elu', name='layer_2'))
encoder.add(L.MaxPooling2D((3,3),padding = 'same',name = 'max_pooling_2'))
encoder.add(L.Conv2D(128, (3,3), padding = 'same', activation = 'elu', name='layer_3'))
encoder.add(L.MaxPooling2D((3,3),padding = 'same',name = 'max_pooling_3'))
encoder.add(L.Conv2D(256, (3,3), padding = 'same', activation = 'elu', name='layer_4'))
encoder.add(L.MaxPooling2D((3,3),padding = 'same',name = 'max_pooling_4'))

encoder.add(L.Flatten())
encoder.add(L.Dense(256))

# decoder
decoder = keras.models.Sequential()
decoder.add(L.InputLayer((code_size,)))
decoder.add(L.Dense(256))
decoder.add(L.Conv2DTranspose(filters=128, kernel_size=(3, 3), strides=2, activation='elu', padding='same'))
decoder.add(L.Conv2DTranspose(filters=64, kernel_size=(3, 3), strides=2, activation='elu', padding='same'))
decoder.add(L.Conv2DTranspose(filters=32, kernel_size=(3, 3), strides=2, activation='elu', padding='same'))
decoder.add(L.Conv2DTranspose(filters=3, kernel_size=(3, 3), strides=2, activation='none', padding='same'))



return encoder, decoder

1 Ответ

0 голосов
/ 13 сентября 2018

В вашем кодере используйте следующее вместо добавления плотного слоя 256: 

decoder.add(L.Dense(2*2*256))             #actual encoder 
decoder.add(L.Reshape((2,2,256)))         #un-flatten
...