Paddle2ONNX 插件的安装与使用¶
PaddleX 的 Paddle2ONNX 插件提供了将 PaddlePaddle 静态图模型转化到 ONNX 格式模型的能力,底层使用Paddle2ONNX。
1. 安装¶
2. 使用¶
2.1 参数介绍¶
参数 | 类型 | 描述 |
---|---|---|
paddle_model_dir | str | 包含Paddle模型的目录。 |
onnx_model_dir | str | ONNX模型的输出目录,可以与Paddle模型目录相同。默认为onnx 。 |
opset_version | int | 使用的ONNX 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 模型为例: