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

星夢妙者
发布: 2025-09-12 08:22:19
原创
273人浏览过

https://www.php.cn/link/e3752bd232f5ce0a575ae0a35c06c79c

近期,YOLOv8官方发布了全新的YOLOv11框架,为当前目标检测领域再添一员得力助手。本文记录了在Windows系统下成功部署YOLOv11的TensorRT模型全过程。

重要提示:环境配置属于基础操作,本文不详细展开安装流程,相关步骤可通过搜索引擎查找对应资源完成。

测试通过环境配置:

  • 操作系统:Windows 10(RTX2070,8GB显存)
  • 开发工具:Visual Studio 2019
  • CMake:3.24.3
  • CUDA:11.7.1 + cuDNN:8.8.0
  • TensorRT:8.6.1.6
  • OpenCV:4.8.0
  • Python环境:Anaconda3 + Python 3.8
  • PyTorch:1.9.0 + cu111
  • Ultralytics:8.3.3

部署流程说明:

部署中最耗时的部分是环境搭建。首先确认系统为Windows 10或11,并配备NVIDIA独立显卡。可通过任务管理器查看(Win10使用

Ctrl+Alt+Delete
登录后复制
,Win11使用
Ctrl+Shift+Esc
登录后复制
),进入“性能”标签页查看GPU信息。

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

立即学习C++免费学习笔记(深入)”;

若显示GPU0、GPU1等设备,且标注为“NVIDIA”,则说明具备可用的独立显卡。若为AMD或集成显卡,则不支持CUDA及TensorRT加速。

接下来需安装以下软件(建议版本一致以快速复现):

  • VS2019 或 VS2022
  • CMake 3.24.3
  • CUDA 11.7.1 + cuDNN 8.8.0
  • TensorRT 8.6.1.6
  • OpenCV 4.8.0
  • Anaconda3 + Python 3.8
  • PyTorch 1.9.0 + cu111
  • Ultralytics 8.3.3

确保上述环境均已正确安装后,开始正式部署。


模型准备与ONNX转换:

前往YOLOv8官方仓库下载YOLOv11模型权重:

? yolo11n.pt 下载地址

将下载的

yolo11n.pt
登录后复制
放入项目目录,在已配置好的YOLO环境(如conda环境)中运行以下命令导出ONNX模型:

yolo export model=yolo11n.pt format=onnx dynamic=False opset=12
登录后复制

或直接运行项目中的

export.py
登录后复制
脚本:

python export.py
登录后复制

成功后将生成

yolo11n.onnx
登录后复制
文件。

AGI-Eval评测社区
AGI-Eval评测社区

AI大模型评测社区

AGI-Eval评测社区 63
查看详情 AGI-Eval评测社区

C++项目编译:

修改项目根目录下的

CMakeLists.txt
登录后复制
,设置OpenCV和TensorRT的安装路径:

# Find and include OpenCV
set(OpenCV_DIR "D:\\lufiles\\opencv480\\build\\x64\\vc16\\lib")
<h1>Set TensorRT path if not set in environment variables</h1><p>set(TENSORRT_DIR "D:\lufiles\TensorRT-8.6.1.6")
登录后复制

随后创建构建目录并生成项目:

mkdir build
cd build
cmake ..
登录后复制

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

进入

build
登录后复制
目录,用Visual Studio打开生成的
.sln
登录后复制
文件。

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

选择 x64 Release 模式,右键点击

ALL_BUILD
登录后复制
项目,选择“生成”。

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

编译成功后,可在

buildRelease
登录后复制
目录下找到生成的
yolov11-tensorrt.exe
登录后复制
可执行文件。


ONNX转TensorRT引擎:

使用TensorRT自带的

trtexec
登录后复制
工具将ONNX模型转换为TensorRT引擎:

trtexec --onnx=yolo11n.onnx --saveEngine=yolo11n.engine --fp16
登录后复制

等待约20分钟,生成

yolo11n.engine
登录后复制
文件。将其复制到
buildRelease
登录后复制
文件夹中。

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


推理测试:

图片检测:

yolov11-tensorrt.exe yolo11n.engine "test.jpg"
登录后复制

视频检测:

yolov11-tensorrt.exe yolo11n.engine "car.mp4"
登录后复制

注意事项:

  • TensorRT引擎与硬件强相关,不可跨设备通用,需在目标机器上重新生成。
  • 如需进行二次开发,建议深入阅读
    main.cpp
    登录后复制
    源码,需具备一定的C++编程基础。
  • 完整源码可从以下链接获取:

? CSDN资源下载链接

以上就是[C++]使用C++部署yolov11目标检测的tensorrt模型支持图片视频推理windows测试通过的详细内容,更多请关注php中文网其它相关文章!

Windows激活工具
Windows激活工具

Windows激活工具是正版认证的激活工具,永久激活,一键解决windows许可证即将过期。可激活win7系统、win8.1系统、win10系统、win11系统。下载后先看完视频激活教程,再进行操作,100%激活成功。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号