软硬件环境
- windows 10 64bit
- pytorch
- yolov5
- deepsort
YOLOv5
前文 YOLOv5目标检测 和 YOLOv5模型训练 已经介绍过了YOLOv5
相关的内容,在目标检测中效果不错。
DeepSort
SORT
算法的思路是将目标检测算法(如YOLO
)得到的检测框与预测的跟踪框的iou
(交并比)输入到匈牙利算法中进行线性分配来关联帧间 ID
。而DeepSORT
算法则是将目标的外观信息加入到帧间匹配的计算中,这样在目标被遮挡但后续再次出现的情况下,还能正确匹配这个ID
,从而减少ID
的切换,达到持续跟踪的目的。