未验证 提交 bfb2dd05 编写于 作者: L LutaoChu 提交者: GitHub

Update model_export.md

上级 004f7dd0
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
通过训练得到一个满足要求的模型后,如果想要将该模型接入到C++预测库或者Serving服务,我们需要通过`pdseg/export_model.py`来导出该模型。 通过训练得到一个满足要求的模型后,如果想要将该模型接入到C++预测库或者Serving服务,我们需要通过`pdseg/export_model.py`来导出该模型。
该脚本的使用方法和`train.py/eval.py/vis.py`完全一样 该脚本的使用方法和`train.py/eval.py/vis.py`完全一样
## FLAGS ## FLAGS
...@@ -12,10 +12,10 @@ ...@@ -12,10 +12,10 @@
## 使用示例 ## 使用示例
我们使用[训练/评估/可视化](./usage.md)一节中训练得到的模型进行试用,脚本如下 我们使用[训练/评估/可视化](./usage.md)一节中训练得到的模型进行试用,命令如下
```shell ```shell
python pdseg/export_model.py --cfg configs/unet_pet.yaml TEST.TEST_MODEL test/saved_models/unet_pet/final python pdseg/export_model.py --cfg configs/unet_optic.yaml TEST.TEST_MODEL ./saved_model/unet_optic/final
``` ```
预测模型会导出到`freeze_model`目录,用于`C++`或者`Python`预测的模型配置会导出到`freeze_model/deploy.yaml` 预测模型会导出到`freeze_model`目录,用于`C++`或者`Python`预测的模型配置会导出到`freeze_model/deploy.yaml`
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册