YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)

笔者使用环境:

        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

下载yolov8s模型

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

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第1张图片

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

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

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第2张图片

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

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

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第3张图片

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第4张图片

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

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第5张图片

依次点击Configure,Generate,Open Project

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第6张图片YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第7张图片

找到main.cpp

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第8张图片

找到源码

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第9张图片

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

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第10张图片

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

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第11张图片

YOLOv8 Tensorrt部署详细介绍(小白从0到1,不会你砍我)_第12张图片

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

你可能感兴趣的:(YOLO)