deep_sort_yolov4
参考我的仓库地址:Github
项目使用darknet版本的yolov4作为detector,不需要将模型进行转换,通过yolo_v4作为前端检测器,后面使用deep_sort官方提供的mars-small128.pb模型和generate_detections.py
代码提取feature,再进行deep_sort算法的跟踪流程。
大致流程:
- yolo_v4根据图片检测物体(人、车辆或其他),获取detections信息包含bbox,confidence等
- 将bbox传入encoder获取feature特征
- 接下来就是deep_sort算法的步骤,开始跟踪
1. 配置yolo_v4:
按博客配置python版本yolo_v4,博文详细提供了如何编译darknet,并测试运行: https://blog.csdn.net/u010512638/article/details/107931119
2. deep_sort:
1)下载MOT16官方数据集MOT16.zip,解压