1 Star 0 Fork 93

NoahSong / PaddleSeg

forked from PaddlePaddle / PaddleSeg 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
model_export_onnx_cn.md 1.39 KB
一键复制 编辑 原始数据 按行查看 历史

简体中文 | English

导出ONNX格式模型

PaddleSeg训练好模型后,也支持导出ONNX格式模型,本教程提供一个示例介绍使用方法。

导出ONNX格式模型的完整方法,请参考Paddle2ONNX

1. 导出预测模型

参考文档导出预测模型。

复用文档中的示例,成功将导出的预测模型文件保存在output文件夹中,如下。

./output
  ├── deploy.yaml            # 部署相关的配置文件,主要说明数据预处理的方式
  ├── model.pdmodel          # 预测模型的拓扑结构文件
  ├── model.pdiparams        # 预测模型的权重文件
  └── model.pdiparams.info   # 参数额外信息,一般无需关注

2. 导出ONNX格式模型

安装Paddle2ONNX(高于或等于0.6版本)。

pip install paddle2onnx

执行如下命令,使用Paddle2ONNX将output文件夹中的预测模型导出为ONNX格式模型。

paddle2onnx --model_dir output \
            --model_filename model.pdmodel \
            --params_filename model.pdiparams \
            --opset_version 11 \
            --save_file output.onnx

导出的ONNX格式模型保存为output.onnx文件。

参考文档:

Python
1
https://gitee.com/noahsong/PaddleSeg.git
git@gitee.com:noahsong/PaddleSeg.git
noahsong
PaddleSeg
PaddleSeg
release/2.7

搜索帮助