paddle 46 使用onnxruntime部署使用paddledetection训练出的旋转框模型(支持batchsize,实现c++与python部署)

57 篇文章 87 订阅 ¥59.90 ¥99.00
本文详细介绍了如何将使用PaddleDetection训练的ppyoloe_r模型导出为ONNX格式,并实现Python和C++的部署。重点在于修改ppyoloe_r_head.py以适配ONNXRuntime,并提供了Python和C++的推理代码示例。
摘要由CSDN通过智能技术生成

本博文以ppyoloe_r为例,实现ppyoloe_r模型的onnx导出部署。
使用paddledetection做旋转框预测可以参考:
paddle 45 使用paddledetection做旋转框预测(预测自己的dota数据集)

1、导出模型

paddledetection中的旋转框模型在导出部署是可以使用paddle框架正常进行部署的,但是由于其在进行旋转框nms时需要特殊的算子,无法在正常的onnxruntime框架进行部署,故此需要修改部分导出源码。同时,ppyoloe_r模型的两个输出shape也没有正常对其,在c++部署时读取数据无法便捷的读取对应grid预测的boxes和class confs。

1.1 修改ppyoloe_r_head.py

修改ppdet/modeling/heads/ppyoloe_r_head.py的post_process函数 只针对paddledetection 2.4有效。新版本的post_process可能需要3个返回值,可以修改return为pred_scores, pred_bboxes,none


                
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

万里鹏程转瞬至

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值