Yolov5 目标检测样例运行

在 ubuntu18.04 系统安装显卡对应驱动，安装好 cuda、cudnn 及与之版本相符的 TensorRT。

接着，我们来进行 yolov5 目标检测：

我们以 omnisky 用户登录，在家目录 /home/omnisky 下建立 work 目录做为工作目录。

mkdir work

cd work

1.下载 yoov5项目

git clone https://github.com/ultralytics/yolov5.git

git clone https://github.com/wang-xinyu/tensorrtx.git

此时，工作目录下有子目录 tensorrtx；

我们将 tensorrtx/yolov5/gen_wts.py 文件拷贝到工作目录下

cp tensorrtx/yolov5/gen_wts.py .

2.下载并转换权重文件

从网上下载权重文件 yolov5s.pt，放到工作目录下

转换权重文件

python gen_wts.py -w yolov5s.pt -o yolov5s.wts

转换后的权重文件是 yolov5s.wts

3.编译 yolov5 项目

cd ~/work/tensorrt/yolov5

mkdir build

cd build

将转换后的权重文件拷贝到该目录下，并编译项目

cp ~/work/yolov5s.wts .

cmake ..

make

4.使用转换后的权重文件生成 .engine 文件

sudo ./yolov5 –s

生成 yolov5s.engine 文件

5.检测，查看效果

我们将图片放到 /home/omnisky/work/picture目录下

sudo ./yolov5 -d ../../picture

检测效果如图：