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



我想要的是获取编码器(压缩数据(的输出,然后对其进行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)

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

autoencoder = Model(input_img, decoded)

来训练它,然后使用单独的类似编码的模型来使用 .predict 方法来获取减少的输入。

相关内容

  • 没有找到相关文章

最新更新