擅长:python、mysql、java
<p>服务器在不同的图形/会话中加载模型,而不是在用于接收客户端请求和响应的图形/会话中加载模型。将<code>MyModel</code>类修改为以下内容:</p>
<pre><code>class MyModel():
def __init__(self):
pass
def load_model(self, model_path):
self.graph = tf.compat.v1.get_default_graph()
with self.graph.as_default():
self.model = tf.keras.models.load_model(model_path)
self.sess = tf.compat.v1.keras.backend.get_session()
def predict(self, img):
with self.graph.as_default():
try:
preds= self.model.predict(img)
except tf.errors.FailedPreconditionError:
tf.compat.v1.keras.backend.set_session(self.sess)
preds= self.model.predict(img)
return preds
</code></pre>