机器学习 keras+flask(或者Django)问题

keras在flask跑出现的问题

alueError: Tensor Tensor("dense_3/Sigmoid:0", shape=(?, 59), dtype=float32) is not an element of this graph.

原始代码

    X = img.reshape([1, 32, width, 1])
    
    y_pred = basemodel.predict(X)
    y_pred = y_pred[:, :, :]
解决方法:

在上面代码所在的py文件前面插入

import tensorflow as tf
global graph,model
graph = tf.get_default_graph()
然在把刚刚的代码改为

    X = img.reshape([1, 32, width, 1])
 
    with graph.as_default():
        y_pred = basemodel.predict(X)
    y_pred = y_pred[:, :, :]
参考博客:https://www.cnblogs.com/svenwu/p/10189557.html

https://blog.csdn.net/lhs960124/article/details/79028691

https://blog.csdn.net/u013465194/article/details/89881317
--------------------- 
作者:北部湾的落日 
来源:CSDN 
原文:https://blog.csdn.net/qq_36213248/article/details/90049915 
版权声明:本文为博主原创文章,转载请附上博文链接!

你可能感兴趣的:(程序报错)