最简单的方法—tensorflow 2.0 keras 保存 .pb 格式的模型
我们知道在tensorflow 2.0中利用 tensorflow.keras可以很方便地保存和加载 .h5 模型,但是 .h5格式的模型只适合在本地使用,不适合部署。tensorflow serving 提供了很强大的部署功能,但是仅支持 .pb 格式,关于 .pb 格式的好处请参考:TensorFlow 保存模型为 PB 文件。本文说明如何用最简单的方法利用 tensorflow.keras ...
原创
2019-12-17 11:01:03 ·
20020 阅读 ·
45 评论