张量流服务使用keras

Question

我想知道如何使用我的模型在生产服务器上使用keras培训。 我听说过tensorflow服务，但我不知道如何使用它与我的keras模型。

我发现这个链接：https://blog.keras.io/keras-as-a-simplified-interface-to-tensorflow-tutorial.html

但我不知道如何初始化SESS变量，因为我的模型已经经过培训和一切。 有没有办法做到这一点？

Answer 1

您可以初始化会话变量

from keras import backend as K 
sess = K.get_session() 

和去导出模型，在教程中（注：进口为出口已经改变）

from tensorflow.contrib.session_bundle import exporter 

K.set_learning_phase(0) 
export_path = ... # where to save the exported graph 
export_version = ... # version number (integer) 

saver = tf.train.Saver(sharded=True) 
model_exporter = exporter.Exporter(saver) 
signature = exporter.classification_signature(input_tensor=model.input, 
               scores_tensor=model.output) 
model_exporter.init(sess.graph.as_graph_def(), 
        default_graph_signature=signature) 
model_exporter.export(export_path, tf.constant(export_version), sess)