当前位置: 代码迷 >> 综合 >> DeepSort+Pychram
  详细解决方案

DeepSort+Pychram

热度:6   发布时间:2023-12-14 02:59:32.0

本文主要是为了深入理解DeepSort而写的,由于时间精力有限慢慢来。

先看以下的两个链接:

code:https://github.com/nwojke/deep_sort

paper: Simple Online and Realtime Tracking with a Deep Association Metric

           Deep Cosine Metric Learning for Person Re-identification

接下来我们看一下代码的构造:

大概是一个这样的结构

--application_util--image_viewer.py #该模块包含一个基于OpenCV的图像查看器和绘图例程--preprocessing.py #预处理--visualization.py #可视化
--deep_sort--detection.py #检测模块--iou_matching.py #该模块包含IOU匹配度量。--kalman_filter.py #用于图像空间过滤的卡尔曼滤波器实现和具体参数化。--linear_assignment.py #用于解决线性分配的问题--nn_matching.py #用于最近邻居匹配度量的模块  emmm..应该是作者自己说的引用的那个啥?哦对余弦测量--track.py #这玩意是单目标追踪器 (哈哈哈我要的就是这玩意)--tracker.py #这玩意是多目标追踪器
--tools  #从冷冻推理图中生成检测--freeze_model.py #查了一下这玩意是推理图--generate_detections.py  #生成检测
--deep_sort_app.py 
--evaluate_motchallenge.py #评估(?)知道的小伙伴告诉我一下这个是什么玩意,评估啥呀
--generate_videos.py #生成视频
--show_results.py  #结果#大概就是这样子了

 

 

1.运行deepsort

创建环境>进入环境

pip install -r requirements.txt

这里有一点要注意,如果想用gpu训练的话,把这个删除

如果安装的话

pip uninstall torch

安装cuda torch

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

点击

pip install requests
pip install pandas
pip install seaborn

运行成功