pytorch yolov5+Deepsort实现目标检测和跟踪+单目测距

本文介绍了如何结合YOLOv5和DeepSort算法进行目标检测和跟踪,作者分享了代码链接和数据,强调了DeepSort的运动轨迹预测和匹配过程。同时提到了卡尔曼滤波和匈牙利算法的应用,但表示部分内容已有优秀资源可供参考。
摘要由CSDN通过智能技术生成

最近一直在整理单目测距的内容,想着检测+单目测距都写完了,顺手也写个检测+跟踪+单目测距,算是总结下这部分内容吧,如果有错误,还请不吝赐教!!
参考文献:
YOLOv5+DeepSort实现目标跟踪
pytorch yolo5+Deepsort实现目标检测和跟踪
感谢大佬们的无私奉献!!

代码链接1
代码链接2
代码链接2对应的数据链接
对代码的部分说明:
1.能跑通最新版本的yolov5就肯定能跑通我的这份代码,就是参考最新代码修改的。
2.权重和数据我就直接放仓库里了,clone下来直接跑,在inference/output里看结果。
3.觉得有用欢迎fork and star!

这两天身体一直不舒服,肠胃不好,本想留着剩下的明天弄,还是挣扎着今天弄完吧,拖延可不是个好习惯!

Deepsort是实现目标跟踪的算法,从sort(simple online and realtime tracking)演变而来,其使用卡尔曼滤波器预测所检测对象的运动轨迹,匈牙利算法将它们与新的检测目标相匹配。Deepsort易于使用且运行速度快,成为AI目标检测跟踪之热门算法。


我发现这篇文章我好像没啥好写的了,卡尔曼滤波可以单独写一篇,匈牙利匹配也可以单独写一篇,没必要合在这里面,其他没啥好写的,参考链接的大佬们都写的很好了,我没必要画蛇添足!


2024年4月14更新:
卡尔曼滤波参考文献,我不可能比他们写的好,所以不写了:
图说卡尔曼滤波,一份通俗易懂的教程
卡尔曼滤波详解
卡尔曼滤波(Kalman Filter)原理与公式推导
控制原理 | 详细总结卡尔曼滤波原理+具体案例分析

匈牙利匹配:
算法学习笔记(5):匈牙利算法


2024年4月18更新:
我这两天总结卡尔曼滤波和匈牙利的知识点,无意中发现了这个视频和工程,我认为作为入门相当合适。卡尔曼很多人都写了,但是这位大佬讲清楚,理顺了知识点,代码写的也很简洁,赞爆了!

单目标:
用卡尔曼滤波器打造一个简易单目标跟踪器
单目标仓库地址:
kalman-filter-in-single-object-tracking
多目标仓库地址:
https://github.com/ZhangPHEngr/Kalman-in-MOT

多目标的结果展示:

  • 6
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是 yolov5+deepsort 行人跟踪环境配置的步骤: 1. 安装 Anaconda,可以从官网下载安装包进行安装。 2. 创建一个 Anaconda 环境,可以使用以下命令: ``` conda create --name <env_name> python=3.7 ``` 其中,`<env_name>` 是你要创建的环境名称。 3. 激活 Anaconda 环境,使用以下命令: ``` conda activate <env_name> ``` 4. 安装 PyTorch,可以根据自己的 CUDA 版本和 PyTorch 版本选择相应的安装方式。下面是一个例子: ``` conda install pytorch torchvision torchaudio cudatoolkit=<cuda_version> -c pytorch ``` 其中,`<cuda_version>` 是你的 CUDA 版本。 5. 安装其它依赖库,可以使用以下命令: ``` pip install opencv-python pillow numpy scipy matplotlib tqdm ``` 6. 克隆 yolov5 代码库,可以使用以下命令: ``` git clone https://github.com/ultralytics/yolov5.git ``` 7. 下载 yolov5 模型文件,可以去 yolov5 代码库的 release 页面下载,也可以使用以下命令: ``` python models/download.py --weights yolov5s.pt ``` 8. 克隆 deepsort 代码库,可以使用以下命令: ``` git clone https://github.com/nwojke/deep_sort.git ``` 9. 下载 deepsort 模型文件,可以去 deepsort 代码库的 release 页面下载,也可以使用以下命令: ``` wget https://github.com/nwojke/deep_sort/releases/download/20200604/deep_sort_model.tar.gz tar -xzvf deep_sort_model.tar.gz ``` 10. 运行行人跟踪程序,可以使用以下命令: ``` python demo.py --source <video_path> --weights yolov5s.pt --cfg deep_sort.yaml --img-size 640 ``` 其中,`<video_path>` 是要处理的视频文件路径。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值