Keras预测过程报错解决: raise ValueError("Tensor %s is not an element of this graph." % obj) ValueError

在使用Keras框架进行图像分割时,第二次开启线程,调用预测类时,在初始化模型阶段会报以下错误:

    raise ValueError("Tensor %s is not an element of this graph." % obj)
ValueError: Tensor Tensor("Placeholder:0", shape=(3, 3, 3, 16), dtype=float32) is not an element of this graph.

很奇怪的是,在第一次调用时不会出错,只有在第二次时会出现这样的错误,猜测是上一次会话未释放的原因。

解决方法如下:

from keras import backend as K

......

K.clear_session()

在预测结束后,及时清除session,和tensorflow框架代码略有区别。

你可能感兴趣的:(机器视觉深度学习,Tensorflow)