跳转至

Paddle2ONNX 插件的安装与使用

PaddleX 的 Paddle2ONNX 插件提供了将 PaddlePaddle 静态图模型转化到 ONNX 格式模型的能力,底层使用Paddle2ONNX

1. 安装

paddlex --install paddle2onnx

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 模型为例:

paddlex \
    --paddle2onnx \
    --paddle_model_dir ./ResNet18 \
    --onnx_model_dir ./ResNet18 \