利用Keras和TensorFlow将模型保存为可部署的PB格式
5星
- 浏览量: 0
- 大小:None
- 文件类型：PDF
简介：
本教程详细介绍如何使用Keras与TensorFlow框架合作，训练并导出深度学习模型至便于服务器端或客户端部署的Protobuf(PB)文件格式。
Keras模型可以保存为可部署的pb格式。要将已训练好的.h5格式的Keras模型转换为.pb文件，请使用以下代码：
```python
import keras
import os
import tensorflow as tf
from tensorflow.python.util import compat
from keras import backend as K
def export_savedmodel(model):
# 传入keras model会自动保存为pb格式
model_path = model # 模型保存的路径
...
```
将训练好的模型传递给`export_savedmodel()`方法即可成功将其保存为.pb文件。
全部评论 (0)
还没有任何评论哟~
