问题描述
Django 使用 Keras 预测模型报错如下:
RuntimeError: The Session graph is empty. Add operations to the graph before calling run().
问题原因
对于这种问题可能有以下可能性:
1、Tensorflow 版本问题
2、Session 问题
解决方法
1、如果是 Tensorflow 版本问题
建议安装 Tensorflow 1.X的版本。
2、Session 问题
我这里是 Session 问题。我这里使用了 Keras 。所以需要引入 Keras 来清理开始会话的 session。
import keras
import tensorflow as tf
import keras.backend.tensorflow_backend as KTF
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
config.gpu_options.per_process_gpu_memory_fraction = 0.7
session = tf.Session(config=config)
KTF.set_session(session)
keras.backend.clear_session() #此句代码是重点