问题说明:
在外部直接调用keras程序不报错,会正常加载模型和权重.但是部署到flask里后就会报错,
The Session graph is empty. Add operations to the graph before calling run().
原因分析:
我的开发环境为tensorflow-gpu-1.14
由于直接运行模型不会报错,所以不应该是keras的配置问题.查了多方资料后看到多个解决办法,底层原因应为flask里的多线程调用会影响session里的图加载导致冲突:
1,有人出现相关问题后,把flask版本降到0.12.2会解决该问题,但是本人测试无效,估计对某一tensorflow版本会起作用
2,flask里多线程调用禁止,如果说开发环境要求不严格只是为测试所用,可以改为此方式应急,但是不能多线程对于实际应用来说不可取
app.run(port=0914, debug=False, host='0.0.0.0', threaded=False)
3,tensorflow升为2.*版本即可解决该问题
4,(该方法未经测试)以下代码在类初始化时候执行一次:
graph = tf.get_default_graph()
sess=tf.Session(graph=graph)