Skip to content

pprp/deep_sort_yolov3_pytorch

 
 

Repository files navigation

DEEP SORT YOLOV3 PYTORCH

论文发表:[1]张宏鸣,汪润,董佩杰,孙红光,李书琴,王红艳.基于LSRCEM-YOLO算法的肉牛多目标跟踪[J/OL].农业机械学报:1-14[2022-03-07].https://kns-cnki-net-s.nudtproxy.yitlink.com:443/kcms/detail/11.1964.S.20210223.0955.004.html.

新特性

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

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

  • 添加了SORT算法

  • 完善ReID部分的训练

快速入门

组织结构

cfg: cfg网络结构文件存放位置

deep_sort

miniversion: 使用cv2中的跟踪模块+yolov3进行跟踪,效果较差

sort: sort算法需要的依赖文件

utils: yolov3中的包

weights: yolov3权重存放位置

deep_sort.py: 仅仅通过运行deep_sort完成目标跟踪过程,保存跟踪的结果(视频文件)

detect.py: 沿用自yolov3,用于侦测目标。

pre_mot.py:进行跟踪,并将结果文件保存下来。

eval_mot.py: 对跟踪的结果文件进行评估,得到指标。

models.py: 沿用自yolov3,是模型构建的代码。

predict.py:沿用自yolov3,侦测单张图片。

sort.py: sort算法再次调用

train.py: 训练yolov3

test.py: 测试yolov3

代码注释

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

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

DeepSort

状态转移:

状态转换图

整体框架:

图片来自知乎Harlek

流程图:

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

参考

目标检测:基于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

About

Replication of MOT project - DeepSort with attention blocks.

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 96.0%
  • Shell 4.0%
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy