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

我在尝试使用Keras构建一个自编码器,输入形状为(470,470,3),但输出始终不匹配,即使我尝试调整填充。这是我写的代码,您能帮我看看吗?目前我的模型摘要显示输出为(472, 472, 3)。

from tensorflow.keras.layers import Conv2D, MaxPooling2D, UpSampling2Dfrom tensorflow.keras import Input, Modelinput_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')

谢谢!


回答:

将最后的填充改为'valid'

decoded_image = Conv2D(3, (3, 3), activation='sigmoid', padding='valid')(x)

Related Posts

KNeighborsRegressor作为去噪算法

在Kaggle上,我发现了用于信号去噪的算法。例如Go…

我只得到`accuracy_score`而不是`roc_auc`的XGBClassifier在GridSearch和交叉验证中

我在使用XGBClassifier处理澳大利亚降雨数据…

可以将AutoML Tables模型导出并在Python中使用吗?

我可以从AutoML Tables中导出一个模型并在本…

如何提高图像分类中相似物体的准确性并降低过拟合

已关闭。此问题属于基于意见的。目前不接受回答。 想要改…

层sequential的输入0与层不兼容:期望ndim=4,但找到ndim=2。接收到的完整形状:[None, 67500]

我在使用CNN模型进行预测时遇到了问题 模型结构 fr…

Numpy NN 在合成数据集上产生奇怪的结果

我正在按照书籍Grokking Deep Learni…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注