pprp / deep_sort_yolov3_pytorch

Replication of MOT project - DeepSort with attention blocks.

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部分的训练

快速入门

• 通过DarkLabel构建DeepSort标注格式和ReID数据集

• Deep SORT多目标跟踪算法代码解析(上)

• Deep SORT多目标跟踪算法代码解析(下)

组织结构

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

deep_sort

• deep: reid模块来自 https://github.com/pprp/reid_for_deepsort
• sort： deep sort沿用了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文件夹中绝大部分代码的注释，以下是根据代码绘制的类图结构：

状态转移：

整体框架：

流程图：

参考

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

deep_sort_yolov3_pytorch's Issues

模型在哪下载？

如题，模型在哪下载？

Why test accuracy is around zero?

When I applied your codes for my own application, I found the training results are reasonable but the test accuracy is too strange. I used the same standard os network as yours and I didn't use center loss. Center loss script is also strange since the losses are all NaN which means the gradients explode. Could you give me some guidance? Thanks in advance.

thanks for your repository. how can i run the code and prepare the dataset? thanks for your help!

视频路径不存在

刚入门小白，我想问下大佬，为什么我在自己的电脑用pycharm跑deepsort.py的时候会显示需要视频路径呢，我在参数设置里加了一个default= ‘’自己的视频路径‘’也还是不可以，大佬可以帮帮我吗。

还有想运行tracker.py的时候，为什么会显示没有父类包，是我缺了什么环境吗

How to train the part of deepsort

请问如何制作ReID数据集并在Deepsort中如何在自己的数据集上对其进行训练，

东西怎么缺了这么多

去哪里找呢？