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 前加個點，引用當(dāng)前目錄下的models

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。