如何将Tensorflow Keras检查点保存到PB文件

Question

有人知道如何将tensorflow keras检查点文件保存到.pb二进制文件中以供服务吗？

我不确定100％如何实现这一目标。

是否需要声明tf.keras.backend.get_session()来保存图形和pb文件？

总而言之， 1.通过tf.keras保存PB文件的方法 要么 2.任何使用tf.keras.estimator.model_to_estimator(keras_model=model)

的方法

谢谢

Answer 1

这里是有关如何使用keras with tensorflow的链接，基本上，您应该从keras获取tensorflow模型（K.get_session（）和K.get_graph（）。graph），然后使用tensorflow methotd保存它。