未验证 提交 49683e76 编写于 作者: S SunAhong1993 提交者: GitHub

Update export_tf_model.md

上级 913c2ac7
...@@ -45,10 +45,10 @@ freeze_model(sess, ["vgg_16/fc8/squeezed"], "vgg16.pb") ...@@ -45,10 +45,10 @@ freeze_model(sess, ["vgg_16/fc8/squeezed"], "vgg16.pb")
``` ```
- 纯checkpoint模型 - 纯checkpoint模型
文件结构: 文件结构:
> |--- checkpoint > |--- checkpoint
> |--- model.ckpt-240000.data-00000-of-00001 > |--- model.ckpt-240000.data-00000-of-00001
> |--- model.ckpt-240000.index > |--- model.ckpt-240000.index
> |--- model.ckpt-240000.meta > |--- model.ckpt-240000.meta
加载和导出模型: 加载和导出模型:
```python ```python
...@@ -81,10 +81,10 @@ freeze_model(sess, ["vgg_16/fc8/squeezed"], save_pb_file) ...@@ -81,10 +81,10 @@ freeze_model(sess, ["vgg_16/fc8/squeezed"], save_pb_file)
- SavedModel模型 - SavedModel模型
文件结构: 文件结构:
> |-- variables > |-- variables
> |------ variables.data-00000-of-00001 > |------ variables.data-00000-of-00001
> |------ variables.data-00000-of-00001 > |------ variables.data-00000-of-00001
> |-- saved_model.pb > |-- saved_model.pb
加载和导出模型: 加载和导出模型:
```python ```python
...@@ -109,4 +109,4 @@ def freeze_model(sess, output_tensor_names, freeze_model_path): ...@@ -109,4 +109,4 @@ def freeze_model(sess, output_tensor_names, freeze_model_path):
# 导出模型 # 导出模型
freeze_model(sess, ["logits"], "model.pb") freeze_model(sess, ["logits"], "model.pb")
``` ```
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册