笔者小白在用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的模型出现的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持w3xue。
经验首页