0


【YOLOv5】yolov5目标识别+DeepSort目标追踪

引言

利用yolov5训练的目标识别模型,结合DeepSort实现目标追踪
源码下载:
(1)Yolov5_DeepSort_Pytorch (该源码下载下来的yolov5文件夹是空的,需要另外下载yolov5的源码)
(2)yolov5
在这里插入图片描述

实现步骤

1 YOLO环境搭建+自定义模型训练

1、参考【YOLOv5-5.0】环境搭建 先将yolov5的环境搭建好
2、参考【YOLO】YOLOv5训练自己的数据集 实现自定义的数据集训练

2 Yolov5_DeepSort_Pytorch 环境搭建

笔者在搭建完YOLO的环境后,基本不用再配置Yolov5_DeepSort_Pytorch,除了个别导入库,利用如下指令进行安装:

cd (自己下载的文件夹)\Yolov5_DeepSort_Pytorch-master
pip install -r requirements.txt

# 测试环境import torch
from IPython.display import Image, clear_output  # 显示结果
 
clear_output()print(f"Setup complete. Using torch {torch.__version__} ({torch.cuda.get_device_properties(0).name if torch.cuda.is_available() else 'CPU'})")# 在运行过程中个别所需的库
pip install 库名 -i https://pypi.douban.com/simple

3 目标追踪实现

需要准备如下文件
(1)yolov5的识别模型(将模型放在.\Yolov5_DeepSort_Pytorch-master\yolov5\weights文件夹下,如图)
(2)重识别/分类模型(将模型放在.\Yolov5_DeepSort_Pytorch-master\deep_sort_pytorch\deep_sort\deep\checkpoint文件夹下,如图)
链接:https://pan.baidu.com/s/1VANSvw3wPvTH53gkNRE0oQ
提取码:6cxt
(3)待追踪的视频(将视频放在.\Yolov5_DeepSort_Pytorch-master文件夹下,如图)
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

在终端运行如下指令:

cd .\Yolov5_DeepSort_Pytorch-master
python track.py --source subway_test.avi --yolo_weights yolov5/weights/Abutment.pt --save-vid

运行过程终端显示如下:
在这里插入图片描述
最终在

.\Yolov5_DeepSort_Pytorch-master\inference\output

生成结果视频:
在这里插入图片描述
在这里插入图片描述
其他的参数设置可以参考

track.py

在这里插入图片描述
综上,实现了基于自定义数据集的目标追踪


本文转载自: https://blog.csdn.net/qq_44703886/article/details/121327643
版权归原作者 摇曳的树 所有, 如有侵权,请联系我们删除。

“【YOLOv5】yolov5目标识别+DeepSort目标追踪”的评论:

还没有评论