Skip to content

Add attention blocks such as cbam, se. Add deep sort, sort and some tracking algorithm using opencv

License

Notifications You must be signed in to change notification settings

vikeezhou/deep_sort_yolov3_pytorch

 
 

Repository files navigation

DEEP SORT YOLOV3 PYTORCH

目标检测部分代码可能有一些问题,将在近期更新

目标检测:基于U版yolov3(版本比较早) https://github.com/ultralytics/yolov3

ReID部分: https://github.com/pprp/reid_for_deepsort

Deep SORT参考: https://github.com/ZQPei/deep_sort_pytorch

SORT参考: https://github.com/abewley/sort

新特性

目标检测部分添加了常用的注意力模块CBAM, SE

添加了使用OpenCV进行目标跟踪的算法,第一帧使用YOLOv3进行检测。(在miniversion文件夹)

添加了SORT算法

完善ReID部分的训练

代码注释

完整讲解《Deep SORT多目标跟踪算法代码解析》在GiantPandaCV公众号首发,欢迎关注。

主要提供了deep_sort文件夹中绝大部分代码的注释,以下是根据代码绘制的类图结构:

DeepSort

状态转移:

状态转换图

整体框架:

图片来自知乎Harlek

流程图:

知乎@猫弟总结的deep sort流程图

About

Add attention blocks such as cbam, se. Add deep sort, sort and some tracking algorithm using opencv

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 96.0%
  • Shell 4.0%