Paddle2ONNX 插件的安装与使用¶
PaddleX 的 Paddle2ONNX 插件提供了将飞桨静态图模型转化到 ONNX 格式模型的能力,底层使用 Paddle2ONNX。
1. 安装¶
2. 使用¶
2.1 参数介绍¶
参数 | 类型 | 描述 |
---|---|---|
paddle_model_dir | str | 包含 Paddle 模型的目录。 |
onnx_model_dir | str | ONNX 模型的输出目录,可以与 Paddle 模型目录相同。默认为 onnx 。 |
opset_version | int | 使用的 ONNX opset 版本。当使用低版本 opset 无法完成转换时,将自动选择更高版本的 opset 进行转换。默认为 7 。 |
2.2 使用方式¶
使用方式:
paddlex \
--paddle2onnx \ # 使用paddle2onnx功能
--paddle_model_dir /your/paddle_model/dir \ # 指定 Paddle 模型所在的目录
--onnx_model_dir /your/onnx_model/output/dir \ # 指定转换后 ONNX 模型的输出目录
--opset_version 7 # 指定要使用的 ONNX opset 版本
以 image_classification 模块中的 ResNet18 模型为例: