news 2026/5/4 1:05:26

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

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
[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++基础才行,否则无法进行二次开发。
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/23 8:16:23

Python+vue3+django的社区桶装饮用水预购管理系统的设计与实现

目录社区桶装饮用水预购管理系统的设计与实现摘要开发技术路线相关技术介绍核心代码参考示例结论源码lw获取/同行可拿货,招校园代理 :文章底部获取博主联系方式!社区桶装饮用水预购管理系统的设计与实现摘要 该系统基于PythonDjangoVue3技术栈开发&…

作者头像 李华
网站建设 2026/4/18 8:53:03

RAG增强与RAG系统技术报告

文章大纲 从架构设计到企业级实践 摘要 第1章 引言 1.1 传统RAG的核心痛点 1.2 PowerRAG的定位 第2章 PowerRAG增强型框架核心能力 2.1 核心技术亮点 2.2 落地效果提升 2.3 与OceanBase一体化融合 第3章 RAG系统全链路技术架构 3.1 两大核心支柱:文档解析 + 检索策略 3.2 嵌入…

作者头像 李华
网站建设 2026/5/1 4:42:10

基于YOLOv8的恶性疟原虫智能检测系统工程实践 [目标检测完整源码]

文章目录基于YOLOv8的恶性疟原虫智能检测系统工程实践 [目标检测完整源码]一、应用背景与问题定义源码下载与效果演示二、整体技术方案设计三、数据集构建与训练策略四、模型推理与系统部署五、工程实现亮点六、应用前景与拓展方向七、结语基于YOLOv8的恶性疟原虫智能检测系统工…

作者头像 李华
网站建设 2026/4/18 8:08:12

写论文软件哪个好?实测揭秘!宏智树 AI 凭 “真研究” 实力出圈

毕业季的论文焦虑,几乎是每一位学子的必经之路:选题太泛被导师打回、文献堆砌逻辑混乱、数据分析无从下手、查重率居高不下、AI 生成痕迹被系统标红…… 面对五花八门的论文写作工具,“写论文软件哪个好” 成了无数学子的灵魂拷问。 作为深耕…

作者头像 李华
网站建设 2026/4/25 18:09:16

持续同调与深度学习3D点云分类方法【附代码】

✨ 本团队擅长数据搜集与处理、建模仿真、程序设计、仿真代码、论文写作与指导,毕业论文、期刊论文经验交流。✅ 专业定制毕设、代码✅ 成品或定制,查看文章底部微信二维码(1)多类型持续同调特征融合模型的构建与验证三维点云作为…

作者头像 李华
网站建设 2026/5/1 7:08:26

基于多特征融合的深度学习高速铁路预售期购票量预测

✨ 本团队擅长数据搜集与处理、建模仿真、程序设计、仿真代码、论文写作与指导,毕业论文、期刊论文经验交流。✅ 专业定制毕设、代码✅ 成品或定制,查看文章底部微信二维码(1)高铁客流数据的时空特征解析与工程化处理 高铁旅客购票…

作者头像 李华