解决Django中调用keras的模型出现的问题
笔者小白在用Django写一个表格单据图片的识别应用的时候,遇到了调用基于Tensorflow的keras模型出错的问题。
出现的错误信息类似于以下:
ValueError:TensorTensor("Placeholder:0",shape=(3,3,1,32),dtype=float32)
通过查询相关的资料,对解决的方式做一个记录。
方法1、通过导入importKeras
然后在构建模型前面加一句keras.backend.clear_session()
方法2、通过提前predict一个全0的数据。
fromkeras.modelsimportload_model importnumpyasnp print('loadmodel...') model=load_model('static\\CnnBankUp.h5',compile=False) print('loaddone.') #一定要添加这段代码,先测试一下,可以避免ValueError:TensorTensor("Placeholder:0",shape=(3,3,1,32),dtype=float32) #isnotanelementofthisgraph.的错误 print('testmodel...') #根据自己传入图片格式定义np.zeros() print(model.predict(np.zeros((2,200,200,1)))) print('testdone.')
以上这篇解决Django中调用keras的模型出现的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。