跳转至

获取 ONNX 模型

PaddleOCR 提供了丰富的预训练模型,这些模型均采用飞桨的静态图格式进行存储。若需在部署阶段使用 ONNX 格式的模型,可借助 PaddleX 提供的 Paddle2ONNX 插件进行转换。关于 PaddleX 及其与 PaddleOCR 之间的关系,请参考 PaddleOCR 与 PaddleX 的区别与联系

首先,执行如下命令,通过 PaddleX CLI 安装 PaddleX 的 Paddle2ONNX 插件:

paddlex --install paddle2onnx

然后,执行如下命令完成模型转换:

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 版本

参数说明如下:

参数 类型 描述
paddle_model_dir str 包含 Paddle 模型的目录。
onnx_model_dir str ONNX 模型的输出目录,可以与 Paddle 模型目录相同。默认为 onnx
opset_version int 使用的 ONNX opset 版本。当使用低版本 opset 无法完成转换时,将自动选择更高版本的 opset 进行转换。默认为 7