获取 ONNX 模型¶
PaddleOCR 提供了丰富的预训练模型,这些模型均采用飞桨的静态图格式进行存储。若需在部署阶段使用 ONNX 格式的模型,可借助 PaddleX 提供的 Paddle2ONNX 插件进行转换。关于 PaddleX 及其与 PaddleOCR 之间的关系,请参考 PaddleOCR 与 PaddleX 的区别与联系。
首先,执行如下命令,通过 PaddleX CLI 安装 PaddleX 的 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 。 |