码农知识堂 - 1000bd
  •   Python
  •   PHP
  •   JS/TS
  •   JAVA
  •   C/C++
  •   C#
  •   GO
  •   Kotlin
  •   Swift
  • paddle 44 用onnxruntime实现ppyoloe模型的部署(含python和c++版本),支持batchsize


    通过正常渠道导出的paddledetection由于内置nms操作,因而无法进行多图推理,正常导出模型部署的代码可以参考paddle 43 用onnxruntime实现ppyoloe模型的部署(含python和c++版本),为此对源码深入研究,实现导出的onnx模型同时推理多个图像,并实现python与C++下的多图推理。

    1、 修改ppyoloe_head.py

    目标检测模型由backbone、neck和head组成,ppyoloe的head在paddledetection源码中为ppyoloe_head.py文件,其head在模型训练与导出时有不同forward路径。在模型导出时默认会boxes解码和nms操作,具体对应的函数为post_process,通过else分支导出的模型会包含nms操作,导致无法支持传入多个图像同时推理(因为nms操作后每个图像的boxes数量都不一致,导致concat操作会失败);只有通过if分支导出的模型,不包含nms才可以正常支持传入多个图像进行推理。为此,需要修改ppyoloe_head.py的代码

    def post_process(self, head_outs
    • 相关阅读:
      重学操作系统(二)进程管理
      Android 四种启动方式
      不同大小的缓冲区对 MD5 计算速度的影响
      vscode连接服务器
      零售数据分析模板分享(通用型)
      K8s中集成Heketi使用Glusterfs
      四个二分万能模板汇总(精简版)
      重温51汇编指令(附实验)
      超好用的PC端录屏软件推荐
      elementUI从接口获取数据,重新组合下拉参数
    • 原文地址:https://blog.csdn.net/a486259/article/details/128132947
    • 最新文章
    • 攻防演习之三天拿下官网站群
      数据安全治理学习——前期安全规划和安全管理体系建设
      企业安全 | 企业内一次钓鱼演练准备过程
      内网渗透测试 | Kerberos协议及其部分攻击手法
      0day的产生 | 不懂代码的"代码审计"
      安装scrcpy-client模块av模块异常,环境问题解决方案
      leetcode hot100【LeetCode 279. 完全平方数】java实现
      OpenWrt下安装Mosquitto
      AnatoMask论文汇总
      【AI日记】24.11.01 LangChain、openai api和github copilot
    • 热门文章
    • 十款代码表白小特效 一个比一个浪漫 赶紧收藏起来吧!!!
      奉劝各位学弟学妹们,该打造你的技术影响力了!
      五年了,我在 CSDN 的两个一百万。
      Java俄罗斯方块,老程序员花了一个周末,连接中学年代!
      面试官都震惊,你这网络基础可以啊!
      你真的会用百度吗?我不信 — 那些不为人知的搜索引擎语法
      心情不好的时候,用 Python 画棵樱花树送给自己吧
      通宵一晚做出来的一款类似CS的第一人称射击游戏Demo!原来做游戏也不是很难,连憨憨学妹都学会了!
      13 万字 C 语言从入门到精通保姆级教程2021 年版
      10行代码集2000张美女图,Python爬虫120例,再上征途
    Copyright © 2022 侵权请联系2656653265@qq.com    京ICP备2022015340号-1
    正则表达式工具 cron表达式工具 密码生成工具

    京公网安备 11010502049817号