当前位置: 首页 > 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服务调用实例及网关安全拦截机制解析
  • 华夏幸福:累计未能如期偿还债务金额合计为227.91亿元
  • 航行警告!南海部分水域进行军事训练,禁止驶入
  • 2025扬州“烟花三月”国际经贸旅游节开幕,37个重大项目现场签约
  • 多米尼加俱乐部屋顶坍塌事故死亡人数升至232人
  • 长三角议事厅|解码中国产业转移新范式:特征识别与韧性构建
  • 睡前玩手机真的很危险,这几种情况一定要小心