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

[C++][cmake]使用C++部署yolo26目标检测的tensorrt模型支持图片视频推理windows测试通过

最近yolo26正式发布了,标志着目标检测又多了一个检测利器,于是尝试在windows下部署yolo26的tensorrt模型,并最终成功。

重要说明:安装环境视为最基础操作,博文不做环境具体步骤,可以百度查询对应安装步骤即可。

测试通过环境:

vs2019

windows 10 RTX2070 8G显存

cmake==3.30.1

cuda11.8.0+cudnn8.9.7

Tensorrt==8.6.1.6

opencv==4.9.0

anaconda3+python3.10

torch==2.5.1+cu124

部署过程:

部署最费时间是安装环境。首先确保自己电脑是win10或者win11并确保电脑有一块nvidia显卡。查看自己显卡就是打开任务管理器(win10是ctrl+alt+delete,win11是ctrl+shift+ESC),在性能里面查看,如下图

如果看到GPU0和GPU1等等表示有显卡,但是需要看到NVIDIA字样才能表示有独立显卡。其他是AMD显卡或者核心显卡,这些都是不能用于cuda的,也就是电脑不支持tensorrt加速和cuda使用的。

首先需要大家安装好VS2019或者VS2022,还有如下环境,由于安装包很多需要去官方搜索下载,需要自己安装,其中版本可以有区别,但是如果快速复现这个项目,最好安装位一致版本这样更快复现出项目。

cmake==3.30.1

cuda11.8.0+cudnn8.9.7

Tensorrt==8.6.1.6

opencv==4.9.0

anaconda3+python3.10

torch==2.5.1+cu124

假设大家安装好上面的环境。下面具体怎么部署,首先去yolo26官方仓库下载yolo26模型,这样下载yolo26n.pt

然后将pt模型放进项目里面,切换自己安装好的yolo26环境里面并切换到项目目录,使用下面命令直接转换

yolo export model=yolo26n.pt format=onnx dynamic=False opset=12

得到onnx模型以后我们开始编译源码。

首先我们修改CMakeLists.txt文件,将源码里面opencv路径和tensorrt路径修改成自己路径

# Find and include OpenCV

set(OpenCV_DIR "D:\\lufiles\\opencv490\\build\\x64\\vc16\\lib")

# Set TensorRT path if not set in environment variables

set(TENSORRT_DIR "D:\\lufiles\\TensorRT-8.6.1.6")

然后执行

mkdir build

cd build

cmake ..

之后去build文件夹找到sln文件用vs打开它,然后选择x64 release,并选中ALL_BUILD右键单击选择生成

之后build\Release文件夹下面有个yolo26-tensorrt.exe生成。之后我们开始转换onnx模型到tensorrt模型,执行命令

trtexec --onnx=yolo26n.onnx --saveEngine=yolo26n.engine --fp16

稍等20多分钟后会自动生成yolo26n.engine文件.

下面我们开始测试图片

yolo26-tensorrt.exe yolo26n.engine "test.jpg"

然后测试视频

yolo26-tensorrt.exe yolo26n.engine "car.mp4"

最终效果如下:

特别注意:

  1. tensorrt模型依赖于硬件,所以不是通用的需要在电脑重新转换,否则可能无法使用;
  2. 如需要二次开发,需要读懂调用代码,需要有一定c++基础才行,否则无法进行二次开发。
http://www.jsqmd.com/news/263248/

相关文章:

  • 【课程设计/毕业设计】基于springboot+微信小程序的驾校在线考试系统驾考在线学习与测试系统【附源码、数据库、万字文档】
  • Python+vue3+django的社区外来人员登记管理系统 流动人口管理系统
  • 2026年国产时序数据库全景盘点:从“精耕细作”到“智能融合”的演进之路
  • Python+vue3+django的社区桶装饮用水预购管理系统的设计与实现
  • 数字员工是什么?熊猫智汇如何提升企业销售效率?
  • 2026必备!9个AI论文平台,自考毕业论文轻松搞定!
  • 2026寒假做题记录
  • Python_django的初中英语学习训练与测评系统
  • 必看!2026年中国十大口碑深海鱼油产品出炉,第一名竟是官方315钦点 - 博客万
  • RLHF模型训练-PPO拆解
  • N-Sum 的算法思想与模板
  • 哪一种辅酶Q10最好?2026辅酶q10十大热门排行榜,为心脏保驾护航 - 博客万
  • 探讨怎样在AI搜索上把企业推广出去,宁波国技互联案例分析 - 工业品牌热点
  • 2026最新贵州大平层装修公司top5榜单发布!贵阳等地装修品牌及施工队综合实力测评 - 品牌推荐2026
  • 2026年剖析AI搜索优化广告,宁波国技互联独特优势大揭秘 - 工业品牌热点
  • 完整教程:从 C 链表到 Android Looper:MessageQueue 的底层原理一条线讲透
  • 国产时序数据库 2026 图鉴:金仓的融合创新与赛道演进方向
  • hdu1059 多重背包
  • hdu1059 多重背包
  • RPM打包进阶:mock与rpmbuild的宏定义传递及spec文件自定义宏实践
  • 2026食品级流量计优选:实力厂家质量保障指南,过热蒸汽流量计/插入式双文丘里/压力变送器,食品级流量计公司怎么选购 - 品牌推荐师
  • 基于Python+django+vue3的高校大学生网上选课网站的设计与实现
  • 2026年宁波地区AI搜索推广公司排名,这些靠谱企业值得关注 - 工业品牌热点
  • 2025年市面上评价高的汽车微动开关实力厂家哪里有,电动推杆微动开关/小型微动开关/防水微动开关直销厂家哪里有 - 品牌推荐师
  • 2026最新贵州改善型装修公司top5榜单发布!贵阳等地装修品牌及施工队综合实力测评,品质工艺双优助力品质家居生活 - 品牌推荐2026
  • 企业大模型微调别乱花钱!从ROI看值不值(附测算工具)
  • 2026最新贵州实景还原家装公司top5榜单发布!贵阳等地装修品牌及施工队综合实力测评,实景还原工艺助力品质家居生活 - 品牌推荐2026
  • 学习unigui【45】UnimDatePicker等按钮汉化崩溃
  • 收集自己的每日早餐类型(包子,豆浆,面包),统计各类型的食材占比,输出营养早餐搭配建议
  • 压箱底的华润万家购物卡别浪费!3 个靠谱渠道盘活沉睡资产 - 可可收