簡體   English   中英

在 keras 中使用自動編碼器的編碼器部分

[英]using the encoder part of an autoencoder in 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方法來獲得減少的輸入。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM