当前位置:   article > 正文

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

yolov8 tensorrt部署

笔者使用环境:

        Tennsorrt8.6.1.6

        CUDA11.7

下载YOLOv8项目和Tensorrt部署项目

  1. git clone https://github.com/xiaocao-tian/yolov8_tensorrt.git
  2. 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需要按照自己的实际情况(其中笔者在这步踩了很多坑,差点劝退,希望大家不要轻言放弃)

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

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

依次点击Configure,Generate,Open Project

找到main.cpp

找到源码

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

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

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

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号