StrongSort (OSNET)配合YOLOv5、v7、v8 实现多目标跟踪详解

  • 此项目包含一个高度可配置的两阶段跟踪器,可根据不同的部署场景进行调整。由YOLOv5, YOLOv7, YOLOv8(一系列在COCO数据集上预训练的对象检测架构和模型)生成的检测被传递给StrongSORT, StrongSORT基于OSNet结合运动和外观信息以跟踪对象。它可以跟踪你的Yolov5模型被训练来检测的任何对象。
  • 本文中yolov7跟踪尚未跑通,但是yolov5、yolov8跟踪及计数功能是直接可用的
  • 其中,yolov5跟踪结果如下图所示,其他结果见文章详解

strongnet + yolov5实现多目标跟踪

环境搭建

#新建虚拟环境
conda create -n v7_sort python=3.9
11-09 09:09