Paddle模型转onnx, PP-OCR系列模型列表

注:看github比看官网文档好(.)


Paddle2ONNX

https://github.com/PaddlePaddle/paddle2onnx/blob/develop/README.md

在这里插入图片描述

Paddle2ONNX模型转化与预测:

https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/deploy/paddle2onnx/readme.md

在这里插入图片描述
在这里插入图片描述

官网: 模型导出ONNX协议

https://www.paddlepaddle.org.cn/documentation/docs/zh/guides/advanced/model_to_onnx_cn.html

一、简介
ONNX (Open Neural Network Exchange) 是针对机器学习所设计的开源文件格式,用于存储训练好的模型。它使得不同的人工智能框架可以采用相同格式存储模型并交互。通过ONNX格式,Paddle模型可以使用OpenVINO、ONNX Runtime等框架进行推理。
Paddle转ONNX协议由 paddle2onnx 实现,下面介绍如何将Paddle模型转换为ONNX模型并验证正确性。

PP-OCR系列模型列表(V3,2022年4月28日更新)

https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/doc/doc_ch/models_list.md

在这里插入图片描述
在这里插入图片描述

前沿算法与模型

https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/doc/doc_ch/algorithm_overview.md

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
当然可以!下面是一个将PaddlePaddle的UNIMO模型换为ONNX格式的示例代码: ```python import paddle from paddle.static import InputSpec import paddlehub as hub import paddle.fluid as fluid # 加载UNIMO模型 module = hub.Module(name="unimo") inputs, outputs, program = module.context(trainable=False) # 定义模型输入 input_spec = [InputSpec(shape=[-1, 128], dtype='int64', name='input_ids'), InputSpec(shape=[-1, 128], dtype='int64', name='segment_ids'), InputSpec(shape=[-1, 128], dtype='int64', name='position_ids'), InputSpec(shape=[-1, 1], dtype='int64', name='input_mask')] # 模型为静态图模式 paddle.enable_static() exe = fluid.Executor(fluid.CPUPlace()) exe.run(fluid.default_startup_program()) # 导出模型参数 fluid.io.save_inference_model(dirname='./unimo_model', feeded_var_names=[x.name for x in inputs], target_vars=outputs, executor=exe, main_program=program) # 将模型参数换为ONNX格式 paddle.onnx.export(model=program, save_file='./unimo.onnx', input_spec=input_spec, opset_version=11) ``` 以上代码首先加载了PaddlePaddle的UNIMO模型,然后定义了模型的输入,接着将模型换为静态图模式,并导出模型参数。最后,使用PaddlePaddleonnx.export函数将模型参数换为ONNX格式,并保存为unimo.onnx文件。 请注意,此示例代码假设你已经安装了PaddlePaddlePaddleHub和ONNX。你可以根据自己的需求修改输入的shape和模型的路径,以及导出的ONNX文件的路径。 希望对你有所帮助!如果还有其他问题,请随时提问。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值