基于YOLOv5+Deepsort或者StrongSort的多目标跟踪(1)

标题 YOLOv5

项目来源:https://github.com/mikel-brostrom/Yolov5_StrongSORT_OSNet/tree/master
基于YOLOv5+Deepsort或者StrongSort的多目标跟踪(1)_第1张图片
改代码采用的是yolov5识别算法和deepsort跟踪算法相结合的方式多类别目标进行跟踪

环境安装

在yolov5代码的基础上就可以运行
或者使用pip install -r requirements.txt 进行安装
基于YOLOv5+Deepsort或者StrongSort的多目标跟踪(1)_第2张图片

代码片段如下

注意需要把跟踪的权重文件放在指定文件夹下面
基于YOLOv5+Deepsort或者StrongSort的多目标跟踪(1)_第3张图片

运行

运行测试视频文件(无跟踪轨迹)

有跟踪轨迹

结果保存在这里
基于YOLOv5+Deepsort或者StrongSort的多目标跟踪(1)_第4张图片
这里的txt文件是需要我们做跟踪评价指标用的
基于YOLOv5+Deepsort或者StrongSort的多目标跟踪(1)_第5张图片

运行MOT16文件,进行跟踪评价指标


注意:跟踪后这个txt文件的名称必须得和跟踪文件的名字保持一致
基于YOLOv5+Deepsort或者StrongSort的多目标跟踪(1)_第6张图片

你可能感兴趣的:(目标跟踪,人工智能,计算机视觉)