有人知道如何将tensorflow keras检查点文件保存到.pb二进制文件中以供服务吗?
我不确定100%如何实现这一目标。
是否需要声明tf.keras.backend.get_session()
来保存图形和pb文件?
总而言之,
1.通过tf.keras
保存PB文件的方法
要么
2.任何使用tf.keras.estimator.model_to_estimator(keras_model=model)
谢谢
答案 0 :(得分:0)
这里是有关如何使用keras with tensorflow的链接,基本上,您应该从keras获取tensorflow模型(K.get_session()和K.get_graph()。graph),然后使用tensorflow methotd保存它。