Skip to content

Latest commit

 

History

History
100 lines (75 loc) · 3.07 KB

model_export_onnx.md

File metadata and controls

100 lines (75 loc) · 3.07 KB

<<<<<<< HEAD

<<<<<<< HEAD English|简体中文

Export model with ONNX format

After training the model by PaddleSeg, we also support exporting model with ONNX format. This tutorial provides an example to introduce it.

For the complete method of exporting ONNX format models, please refer to Paddle2ONNX。

1.Export the inference model

Refer to document to export model, and save the exported inference model to the output folder, as follows.

./output
  ├── deploy.yaml            # deployment-related profile
  ├── model.pdmodel          # topology file of inference model
  ├── model.pdiparams        # weight file of inference model
  └── model.pdiparams.info   # additional information, generally do not need attention to this file

2. Export ONNX format model

Install Paddle2ONNX (version 0.6 or higher).

PaddlePaddle-release/2.4

导出ONNX格式模型

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

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

1. 导出预测模型

参考文档导出预测模型。

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

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

2. 导出ONNX格式模型

安装Paddle2ONNX（高于或等于0.6版本)。 <<<<<<< HEAD

9c8570af (add new models) PaddlePaddle-release/2.4

pip install paddle2onnx

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

======= <<<<<<< HEAD Execute the following command to export the prediction model in the output folder to an ONNX format model by Paddle2ONNX.

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

9c8570af (add new models) PaddlePaddle-release/2.4

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

<<<<<<< HEAD

<<<<<<< HEAD The exported ONNX format model is saved as output.onnx file.

Reference documents:

=======

PaddlePaddle-release/2.4 导出的ONNX格式模型保存为output.onnx文件。

参考文档：

9c8570af (add new models) PaddlePaddle-release/2.4