python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > keras+flask模型的重复调用出错ValueError

解决keras+flask模型的重复调用出错ValueError: Tensor is not an element of this graph

作者:七彩吞天蟒

这篇文章主要介绍了解决keras+flask模型的重复调用出错ValueError: Tensor is not an element of this graph问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

确认基调:这个问题不是大问题,看完本篇文章就能解决

1. 错误重现

ValueError: Tensor Tensor("dense_3/BiasAdd:0", shape=(?, 1), dtype=float32) is not an element of this graph.

2. 错误含义

表面含义:传入的张量[tensor]并不是这个图[graph]的元素。具体意义:就是说,当你进行模型预测的时候,传入的数据,和后台的图不匹配。

3. 解决方法

方法一

把keras的tensorflow的后端换成theano,具体操作不赘述

这个方法不推荐,因为并没有实际解决问题,而是逃避问题,而且有可能会出现一些新的问题。【嗯,我就出现了】

方法二

选择每次调用模型的时候,重新加载一下模型,也就是说把加载模型写在调用的函数里,这样模型就是个局部变量,每次调用每次加载。【如果还是报错】

那记得在模型预测结果后,在其代码追加一句:

ret = model.predict(x_test)
keras.backend.clear_session()

意思就是清理下session,这样每次调用都会加载一下模型,并且清除session

这个方法也不推荐,加载模型耗时耗力,而且清除了其他的session,有可能造成其他程序的崩溃

方法三

创建全局的model,创建全局的graph,调用的时候直接引用过来

像这样:

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)

但是这不算结束,关键的地方来了,在调用模型的时候,要引用一下全局的图才行

像这样:

with graph.as_default():
	prd = model.predict(x_test)

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文