Tensor(*** shape=(?, 1, ?, 5693), dtype=float32) is not an element of this graph.

天马行空波

于 2020-06-22 20:29:34 发布

阅读量778

点赞数 1

分类专栏： python

本文链接：https://blog.youkuaiyun.com/jiangbo721/article/details/106909216

版权

29 篇文章

订阅专栏

本文探讨了Flask框架与TensorFlow在多线程环境下出现的兼容性问题，详细解析了两种解决方法：一是禁用gevent异步处理，二是通过调用model._make_predict_function()确保模型在所有线程中正确加载。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

因为主线程是子线程的关系，以及flask使用了gevent异步，导致在其他线程没有主线程里的模型，最终导致识别报错。

解决的方法也很简单

取消代码中的gevent

# from gevent import monkey;
# monkey.patch_all()

还有启动时候取消gevent

gunicorn -w ${WORKER} -t 100 -k gevent -b 0.0.0.0:${PORT} server:app
gunicorn -w ${WORKER} -t 100 -b 0.0.0.0:${PORT} server:app  # 去掉gevent

就可以了

在load_model 之后加上
model._make_predict_function()
参见：https://github.com/keras-team/keras/issues/6462
这段代码即可。
搞定！