Django呼叫keras的模型出現問題的解決方法

2021-08-25 22:40:40 字數 971 閱讀 4950

筆者小白在用django寫乙個**單據的識別應用的時候,遇到了呼叫基於tensorflow的keras模型出錯的問題。

出現的錯誤資訊類似於以下:

valueerror: tensor tensor("placeholder:0", shape=(3, 3, 1, 32), dtype=float32)
通過查詢相關的資料,對解決的方式做乙個記錄。

方法1、通過匯入import keras

然後在構建模型前面加一句keras.backend.clear_session()

方法2、通過提前predict乙個全0的資料。

from keras.models import load_model

import numpy as np

print('load model...')

model = load_model('static\\cnnbankup.h5', compile=false)

print('load done.')

#一定要新增這段**,先測試一下,可以避免valueerror: tensor tensor("placeholder:0", shape=(3, 3, 1, 32), dtype=float32)

#is not an element of this graph.的錯誤

print('test model...')

#根據自己傳入格式定義np.zeros()

print(model.predict(np.zeros((2, 200,200,1))))

print('test done.')

參考文獻:

1、 2018.8.26

2、 2018.8.26

3、 2018.8.26

Keras學習 1 使用keras建立序列模型

keras學習 1 使用keras建立序列模型 sequential model就是一些列layers的簡單堆疊。首先,我們建立乙個簡單的前向全連線網路。輸入維度784 from keras.models import sequential from keras.layers import dens...

TensorFlow呼叫Keras庫函式存在的問題

tensorflow在1.4版本引入了keras,封裝成庫。現想將keras版本的gru 移植到tensorflow中,看到tensorflow中有keras庫,大喜,故將神經網路定義部分使用keras的function api方式進行定義,訓練部分則使用tensorflow來進行編寫。一頓操作之後...

cppIntergration 呼叫 cu模組

main.cpp extern c bool runtest const int argc,const char ar char data,int2 data int2,unsigned int len 在主程式裡使用extern c 將.cu函式擴充套件到所有檔案可用cppintegration....