YOLOv8 Tensorrt部署详细介绍（小白从0到1，不会你砍我）

Candy_Rainbow_

已于 2023-10-09 11:31:56 修改

阅读量2.2k

点赞数 4

文章标签： YOLO

于 2023-10-08 15:29:37 首次发布

本文链接：https://blog.csdn.net/Candy_Rainbow_/article/details/133669218

版权

笔者使用环境：

Tennsorrt8.6.1.6

CUDA11.7

下载YOLOv8项目和Tensorrt部署项目

git clone https://github.com/xiaocao-tian/yolov8_tensorrt.git
git clone https://github.com/ultralytics/ultralytics.git

在YOLOv8项目中新建weights文件夹，将yolov8s.pt放入

运行tensorrt项目中gen_wts.py文件，将pt文件转成wts文件，如上图

在tensorrt项目中新建weights文件夹，将wts文件复制过来，如下图

在tensorrt项目中新建build文件夹，然后使用cmake编译，注意tensorrt项目中Cmakelist.txt需要按照自己的实际情况（其中笔者在这步踩了很多坑，差点劝退，希望大家不要轻言放弃）

set设置为本地电脑上的OpenCV和Tensorrt绝对路径，例如：

修改其中两个路径即可，接着打开cmake

依次点击Configure，Generate，Open Project

找到main.cpp

找到源码

其中将 wts_name = “” 注释掉就是生成engine文件，取消注释便是推理

先取消注释得到engine文件，如上图所示（此过程大概需要十分钟）

打开注释，修改自己待检测的视频路径，本地调试，运行成功

关注