解决Django中调用keras的模型出现的问题

雪花,是世界上最圣洁的花,当你静静的倾听天籁,尘世的浮躁与喧嚣,似乎都已离你远去,剩下的惟有纯净,那是心灵忘却一切的畅然,那是在没有尘埃的世界里,生命自由呼吸的平淡与安恬。

笔者小白在用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.')

以上这篇解决Django中调用keras的模型出现的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

您可能有感兴趣的文章
django连接数据库获取数据的简单步骤记录

Django实现翻页的示例代码

正确的理解和使用Django信号(Signals)

django使用多个数据库的方法实例

Django中如何用xlwt生成表格的方法步骤