解決keras+flask模型的重復(fù)調(diào)用出錯(cuò)ValueError: Tensor is not an element of this graph
確認(rèn)基調(diào):這個(gè)問題不是大問題,看完本篇文章就能解決
1. 錯(cuò)誤重現(xiàn)
ValueError: Tensor Tensor("dense_3/BiasAdd:0", shape=(?, 1), dtype=float32) is not an element of this graph.
2. 錯(cuò)誤含義
表面含義:傳入的張量[tensor]并不是這個(gè)圖[graph]的元素。具體意義:就是說,當(dāng)你進(jìn)行模型預(yù)測(cè)的時(shí)候,傳入的數(shù)據(jù),和后臺(tái)的圖不匹配。
3. 解決方法
方法一
把keras的tensorflow的后端換成theano,具體操作不贅述
這個(gè)方法不推薦,因?yàn)椴]有實(shí)際解決問題,而是逃避問題,而且有可能會(huì)出現(xiàn)一些新的問題?!距牛揖统霈F(xiàn)了】
方法二
選擇每次調(diào)用模型的時(shí)候,重新加載一下模型,也就是說把加載模型寫在調(diào)用的函數(shù)里,這樣模型就是個(gè)局部變量,每次調(diào)用每次加載?!救绻€是報(bào)錯(cuò)】
那記得在模型預(yù)測(cè)結(jié)果后,在其代碼追加一句:
ret = model.predict(x_test) keras.backend.clear_session()
意思就是清理下session,這樣每次調(diào)用都會(huì)加載一下模型,并且清除session
這個(gè)方法也不推薦,加載模型耗時(shí)耗力,而且清除了其他的session,有可能造成其他程序的崩潰
方法三
創(chuàng)建全局的model,創(chuàng)建全局的graph,調(diào)用的時(shí)候直接引用過來
像這樣:
if __name__ == '__main__':
graph = tf.get_default_graph()
model = keras.models.load_model('./data/model/model.model')
app.run(host='0.0.0.0', port=9090, debug=True)
但是這不算結(jié)束,關(guān)鍵的地方來了,在調(diào)用模型的時(shí)候,要引用一下全局的圖才行
像這樣:
with graph.as_default(): prd = model.predict(x_test)
總結(jié)
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
在Django中同時(shí)使用多個(gè)配置文件的方法
這篇文章主要介紹了在Django中同時(shí)使用多個(gè)配置文件的方法,Django是豐富多彩的Python web框架中人氣最高的一個(gè),需要的朋友可以參考下2015-07-07
一文教你用python編寫Dijkstra算法進(jìn)行機(jī)器人路徑規(guī)劃
迪杰斯特拉(Dijkstra)算法是典型最短路徑算法,用于計(jì)算一個(gè)節(jié)點(diǎn)到其他節(jié)點(diǎn)的最短路徑,這篇文章主要給大家介紹了關(guān)于利用python編寫Dijkstra算法進(jìn)行機(jī)器人路徑規(guī)劃的相關(guān)資料,需要的朋友可以參考下2021-08-08
python:目標(biāo)檢測(cè)模型預(yù)測(cè)準(zhǔn)確度計(jì)算方式(基于IoU)
今天小編就為大家分享一篇python:目標(biāo)檢測(cè)模型預(yù)測(cè)準(zhǔn)確度計(jì)算方式(基于IoU),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-01-01
python?matplotlib庫(kù)繪圖實(shí)戰(zhàn)之繪制散點(diǎn)圖
Python有著強(qiáng)大的繪圖庫(kù) matplotlib,該庫(kù)集成了大量的繪制函數(shù),可以滿足我們平時(shí)絕大多數(shù)的繪圖要求,這篇文章主要給大家介紹了關(guān)于python?matplotlib庫(kù)繪圖實(shí)戰(zhàn)之繪制散點(diǎn)圖的相關(guān)資料,需要的朋友可以參考下2022-07-07
DataFrame.to_excel多次寫入不同Sheet的實(shí)例
今天小編就為大家分享一篇DataFrame.to_excel多次寫入不同Sheet的實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-12-12
Python中return函數(shù)返回值實(shí)例用法
在本篇文章里小編給大家整理的是一篇關(guān)于Python中return函數(shù)返回值實(shí)例用法,有興趣的朋友們可以學(xué)習(xí)下。2020-11-11
python回調(diào)函數(shù)中使用多線程的方法
這篇文章主要介紹了python回調(diào)函數(shù)中使用多線程的方法,需要的朋友可以參考下2017-12-12

