【TFServing】Keras模型导出为SavedModel（支持TF1.X和TF2.X版本）

Keras模型导出

可以用TensorFlow自带的函数，方便的将Keras模型导出为 SavedModel 格式，部署到TF Serving上。具体代码如下：
【提示】因版本差异，TensorFlow 1.X的版本和TensorFlow 2.X的版本导出的代码有略微不同。

---

TensorFlow 1.X 版本的导出代码

import tensorflow as tf

tf.saved_model.save(
    obj=model,
    export_dir='./model_name/1',
    signatures=None)

TensorFlow 2.X 版本的导出代码

import tensorflow as tf

tf.compat.v1.saved_model.save(
    obj=model,
    export_dir='./model_name/1',
    signatures=None)

保存结果展示

保存好的 SavedModel 格式的 Keras模型 的输出目录如下：

|--model_name
|    |--1
|        |--saved_model.pb
|        |--variables
|            |--variables.data-00000-of-00001
|            |--variables.index

会在指定目录下面生成一个pb文件，以及一个variables文件夹。就说明Keras模型导出成功了。

---

参考资料

• https://zhuanlan.zhihu.com/p/128546377
• https://blog.csdn.net/qq_36643449/article/details/124609848