在keras中使用自动编码器的编码器部分

问题描述 投票:-1回答:1

我想要的是得到编码器(压缩数据)的输出,然后对它进行face_recognition。训练完这个自动编码器后,我想使用经过训练的编码器。

因此,当我尝试运行代码时,我得到了这个错误:我如何解决问题并仅提取此自动编码器模型的训练编码器部分?

ValueError: Error when checking target: expected max_pooling2d_3 to have shape (8, 8, 64) but got array with shape (64, 64, 3)
python keras deep-learning face-recognition autoencoder
1个回答
0
投票

发生的事情是您的模型输出是encoded部分,并且您将提供您将编码为目标的图像,这对于自动编码器是正确的。你需要做的是定义

autoencoder = Model(input_img, decoded)

训练它然后使用一个单独的编码模型来使用.predict方法来获得减少的输入。

© www.soinside.com 2019 - 2024. All rights reserved.