App下載

Keras加載model出錯怎么辦?

猿友 2021-08-04 16:16:15 瀏覽數(shù) (3435)
反饋

Keras是一個出名的機器學習框架,我們用這個框架訓練一個model可以用來進行預(yù)測或者一些學習操作。但是在使用這個model的時候有時候會報錯,比如有時候Keras加載model的時候就會報錯。這種情況常見于反復(fù)調(diào)用的情況。那么如何解決Keras加載model報錯呢?接下來的這篇文章告訴你。

問題:

python+Flask 反復(fù)調(diào)用model報錯

ValueError: Tensor Tensor("dense_2/Softmax:0"...) is not an element of this graph.

原因:

不明

解決方案:

在初始化加載模型之后,就隨便生成一個向量讓 model 執(zhí)行一次 predict 函數(shù)

from keras.models import load_model
# ...
model = load_modle('xxx.hdf5')
model.predict(...)

補充:from models import * 報錯,找不到models的解決辦法

解決方法

from .models import *

在models 前加個點,引用當前目錄下的models

以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持W3Cschool


0 人點贊