尝试将我的keras模型导出到tensorflow服务时出现错误

时间:2018-12-15 17:56:00

标签: python tensorflow keras tensorflow-serving

from tensorflow.python.saved_model import builder as saved_model_builder
export_path=r'tensorflowservingmodel'
export_version=1
builder = saved_model_builder.SavedModelBuilder(export_path)
signature = predict_signature_def(
    inputs={'input': new_model.inputs[0]},
    outputs={'output': new_model.outputs[0]})


with K.get_session() as sess:

    builder.add_meta_graph_and_variables(
        sess=sess,
        tags=[tag_constants.SERVING],
        clear_devices = True,
        signature_def_map={
            signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY: signature}
    )
builder.save()
print("tensorflow serving.......")
  

AssertionError Traceback(最近的呼叫   最后)        2 export_path = r'tensorflowservingmodel'        3 export_version = 1 ----> 4 builder = saved_model_builder.SavedModelBuilder(export_path)        5个签名= predict_signature_def(        6个输入= {'input':new_model.inputs [0]},          〜\ Anaconda \ lib \ site-packages \ tensorflow \ python \ saved_model \ builder_impl.py   在 init (自己,export_dir)中       93提高AssertionError(       94“出口目录已存在。请指定其他出口” ---> 95“目录:%s”%export_dir)       96       97 file_io.recursive_create_dir(self._export_dir)          AssertionError:导出目录已存在。请指定另一个导出目录:tensorflowservingmodel

2 个答案:

答案 0 :(得分:1)

删除此目录“ tensorflowservingmodel”,在导出模型之前确保它不存在。

rm -r tensorflowservingmodel

答案 1 :(得分:0)

我发现问题是我已经创建了一个名为tensorflowservingmodel的文件,这就是程序说目录已经存在的原因。我不得不删除文件目录并重新运行代码。