[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.mzph.cn/news/1178691.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

【课程设计/毕业设计】基于springboot+微信小程序的驾校在线考试系统驾考在线学习与测试系统【附源码、数据库、万字文档】

博主介绍:✌️码农一枚 ,专注于大学生项目实战开发、讲解和毕业🚢文撰写修改等。全栈领域优质创作者,博客之星、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java、小程序技术领域和毕业项目实战 ✌️技术范围:&am…

Python+vue3+django的社区外来人员登记管理系统 流动人口管理系统

目录社区外来人员登记管理系统摘要开发技术路线相关技术介绍核心代码参考示例结论源码lw获取/同行可拿货,招校园代理 :文章底部获取博主联系方式!社区外来人员登记管理系统摘要 随着城市化进程加快,流动人口数量激增,传统人工登记…

2026年国产时序数据库全景盘点:从“精耕细作”到“智能融合”的演进之路

摘要:进入2026年,在“数字中国”与工业物联网浪潮的强劲推动下,国产时序数据库市场持续繁荣,竞争格局日趋清晰。本文将对当前主流的国产时序数据库进行梳理盘点,并特别聚焦于金仓数据库(Kingbase&#xff0…

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

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

数字员工是什么?熊猫智汇如何提升企业销售效率?

数字员工通过自动化其销售和客户服务任务,显著优化了企业的业务流程。这些智能化工具不仅能快速处理大量日常询问,还能进行数据分析,从而实时反馈客户需求。这种高效性让企业能迅速响应市场变化,减少人工操作,提高了工…

2026寒假做题记录

不断更新中... Codeforces Round 1069 (Div. 2) A. Little Fairys Painting 开桶,模拟过程即可,发现 \(c_i\) 不变便跳出模拟,一定不会超时。 B. XOR Array 利用异或前缀和,设 \(b_i=a_1 \xor a_2\^{}...\^{}a_i\)…

2026必备!9个AI论文平台,自考毕业论文轻松搞定!

2026必备!9个AI论文平台,自考毕业论文轻松搞定! AI 工具让论文写作不再难 随着人工智能技术的不断进步,越来越多的学生开始借助 AI 工具来提升论文写作效率。尤其是在自考过程中,面对繁重的论文任务,如何在…

Python_django的初中英语学习训练与测评系统

目录初中英语学习训练与测评系统(基于Python Django)开发技术路线相关技术介绍核心代码参考示例结论源码lw获取/同行可拿货,招校园代理 :文章底部获取博主联系方式!初中英语学习训练与测评系统(基于Python Django&…

必看!2026年中国十大口碑深海鱼油产品出炉,第一名竟是官方315钦点 - 博客万

随着生活节奏加快和饮食结构变化,高血脂、心血管问题逐渐成为困扰大众的健康隐患。深海鱼油因富含Omega-3脂肪酸,对调节血脂、保护心血管具有积极作用,一跃成为保健品市场的“明星产品”。然而,市场上深海鱼油品牌…

RLHF模型训练-PPO拆解

零、基本概念 在一个RLHF的流程中,包含了trainer 和 rollout两个过程,其中trainer是训练强化学习的部分,而rollout则是执行模型推理的部分。 在PPO策略中,trainer的主要目的则是为了训练策略模型,它包括了策略模型…

N-Sum 的算法思想与模板

终结 N-Sum 的算法思想与模板:以 3-Sum 和 4-Sum 为例 在算法面试和 LeetCode 中,N-Sum 问题是一个经典的考察点。无论是 3-Sum,还是 4-Sum,这些问题都是基于同一个核心思想:排序 + 双指针收缩。本文将通过 3-Sum…

2026最新贵州装修设计公司top5榜单发布!贵阳等地装修品牌及施工队综合实力测评,工艺与口碑双优助力品质家居生活 - 品牌推荐2026

引言 随着家居消费升级,消费者对装修设计的实用性、耐用性与个性化需求日益提升,但行业存在设计同质化、工艺不透明、售后无保障等问题,导致装修体验大打折扣。据中国建筑装饰协会2025年度报告显示,全国家装行业投…

哪一种辅酶Q10最好?2026辅酶q10十大热门排行榜,为心脏保驾护航 - 博客万

哪一种辅酶Q10最好?2026辅酶q10十大热门排行榜,为心脏保驾护航 打开购物软件,辅酶Q10的热门推荐刷不完,每款都标注“护心优选”“热门爆款”,评论区更是清一色好评。可真正买回家尝试,要么没感受到任何正向反馈,…

2026最新贵州大平层装修公司top5榜单发布!贵阳等地装修品牌及施工队综合实力测评 - 品牌推荐2026

引言 随着改善型住房需求持续增长,大平层装修市场迎来品质升级新浪潮,但行业存在设计同质化、工艺不透明、售后无保障等问题,消费者选择难度显著增加。据中国室内装饰协会2026年第一季度数据显示,全国大平层装修投…

探讨怎样在AI搜索上把企业推广出去,宁波国技互联案例分析 - 工业品牌热点

在AI技术重塑商业生态的今天,如何通过AI搜索为企业做推广、怎样在AI搜索上把企业推广出去,已成为中小企业突破增长瓶颈的关键命题。宁波国技互联作为深耕数字经济领域的先行者,凭借独创的GEO-AI搜索体系,为企业提供…

2026年剖析AI搜索优化广告,宁波国技互联独特优势大揭秘 - 工业品牌热点

在AI技术深度融入商业生态的今天,一套高效的AI搜索优化方案是企业打通智能营销链路、实现精准获客的核心抓手。面对市场上良莠不齐的AI搜索服务提供商,如何找到既懂技术又懂行业的合作伙伴?以下结合不同服务类型,为…

完整教程:从 C 链表到 Android Looper:MessageQueue 的底层原理一条线讲透

pre { white-space: pre !important; word-wrap: normal !important; overflow-x: auto !important; display: block !important; font-family: "Consolas", "Monaco", "Courier New", …

hdu1059 多重背包

import java.util.Scanner;public class hdu1059 {//01背包public static boolean bag(int[] weight,int[] value,int V){int[] res = new int[V+1];for (int i = 0; i < weight.length; i++) {for (int j = V; j &…

国产时序数据库 2026 图鉴:金仓的融合创新与赛道演进方向

前言 进入2026年&#xff0c;在“数字中国”与工业物联网浪潮的强劲推动下&#xff0c;国产时序数据库市场持续繁荣&#xff0c;竞争格局日趋清晰。本文将对当前主流的国产时序数据库进行梳理盘点&#xff0c;并特别聚焦于金仓数据库&#xff08;Kingbase&#xff09;&#xff…

hdu1059 多重背包

import java.util.Scanner;public class hdu1059 {//01背包public static boolean bag(int[] weight,int[] value,int V){int[] res = new int[V+1];for (int i = 0; i < weight.length; i++) {for (int j = V; j &…