Keras自动编码器的输入和输出层不匹配,无法运行模型

2024-06-23 19:36:44 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在Keras中构建一个自动编码器,输入形状为(470470,3),但输出似乎永远不匹配,即使我尝试切换填充。这是我的密码,你能帮我吗?我的模型摘要当前的编写方式显示了(472472,3)的输出

from tensorflow.keras.layers import Conv2D, MaxPooling2D, UpSampling2D
from tensorflow.keras import Input, Model

input_image = Input(shape=(470, 470, 3)) 

x = Conv2D(32, (3, 3), activation='relu', padding='same')(input_image)
x = MaxPooling2D((2, 2), padding='same')(x)
x = Conv2D(32, (3, 3), activation='relu', padding='same')(x)
encoded = MaxPooling2D((2, 2), padding='same')(x)

x = Conv2D(32, (3, 3), activation='relu', padding='same')(encoded)
x = UpSampling2D((2, 2))(x)
x = Conv2D(32, (3, 3), activation='relu', padding='same')(x)
x = UpSampling2D((2, 2))(x)
decoded_image = Conv2D(3, (3, 3), activation='sigmoid', padding='same')(x)

autoencoder = Model(input_image, decoded_image)
autoencoder.compile(optimizer='adadelta', loss='binary_crossentropy')

谢谢大家!


Tags: fromimageimportinputmodeltensorflowactivationkeras

热门问题