当前位置: 首页 > news >正文

pytorch 51 GroundingDINO模型导出tensorrt并使用c++进行部署,53ms一张图

本专栏博客第49篇文章分享了将 GroundingDINO模型导出onnx并使用c++进行部署,并尝试将onnx模型转换为trt模型,fp16进行推理,可以发现推理速度提升了一倍。为此对GroundingDINO的trt推理进行调研,发现
在GroundingDINO-TensorRT-and-ONNX-Inference项目中分享了模型导出onnx转trt格式,并分享了python-trt推理代码,但没有c++推理代码。为此,博主基于https://github.com/hpc203/GroundingDINO-onnxrun项目分享的onnx c++推理代码,修改为GroundingDINO模型的tensorrt推理代码。对于600x400的输入,在3060显卡下,cuda 12.1,TensorRT-10.4.0,window环境下53ms处理完一张图片。

1、前置操作

1.1 trt模型导出

参考

相关文章:

  • 大数据建模与评估
  • Linux `init 4` 相关命令的完整使用指南
  • Nvidia显卡架构演进
  • 新手记录--从零开始[labelme安装及使用]
  • Ubuntu20.04下Docker方案实现多平台SDK编译
  • 分析虚幻引擎编辑器中使用 TAA 或 TSR 时角色眨眼导致的眼睛模糊问题
  • 【java实现+4种变体完整例子】排序算法中【归并排序】的详细解析,包含基础实现、常见变体的完整代码示例,以及各变体的对比表格
  • 洛谷P1312 [NOIP 2011 提高组] Mayan 游戏
  • 【AI】解决分类问题
  • SICAR 标准 安全门锁操作箱 按钮和指示灯说明
  • 前端vue+后端ssm项目
  • 桌面应用UI开发方案
  • Linux第一个系统程序——进度条
  • UIjavaScritIU
  • 模拟投资大师思维:AI对冲基金开源项目详解
  • 从零搭建微服务项目Pro(第6-2章——微服务鉴权模块SpringSecurity+JWT)
  • React-memo (useMemo, useCallback)
  • javassist
  • windows下用xmake交叉编译鸿蒙.so库
  • 【Easylive】Interact与Web服务调用实例及网关安全拦截机制解析
  • 《水饺皇后》领跑五一档票房,《哪吒2》上座率仍居第一
  • 李强签署国务院令,公布修订后的《中华人民共和国植物新品种保护条例》
  • 解放日报:浦东夯实“热带雨林”式科创生态
  • 民生访谈|支持外贸企业拓内销,上海正抓紧制定便利措施
  • 欢迎回家!神十九返回舱成功着陆
  • 外媒称菲方允许菲官员窜台,国台办:应停止在台湾问题上玩火