下载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放入
运行tensorrt项目中gen_wts.py文件,将pt文件转成wts文件,如上图
在tensorrt项目中新建weights文件夹,将wts文件复制过来,如下图
在tensorrt项目中新建build文件夹,然后使用cmake编译,注意tensorrt项目中Cmakelist.txt需要按照自己的实际情况(其中笔者在这步踩了很多坑,差点劝退,希望大家不要轻言放弃)
修改其中两个路径即可,接着打开cmake
依次点击Configure,Generate,Open Project
找到main.cpp
找到源码
其中将 wts_name = “” 注释掉就是生成engine文件,取消注释便是推理
先取消注释得到engine文件,如上图所示(此过程大概需要十分钟)
打开注释,修改自己待检测的视频路径,本地调试,运行成功