張量流服務使用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)