Wei-JL/deep_sort_paddle

Deep Sort算法的Paddle实现

简介

• 注：只是基于原作者做一下修改，点我进入原地址：模型开发

• 基于DeepSort官方开源代码开发，将其中的深度学习模型更换为Paddle模型

• 预训练检测模型来自PaddleDetection套件中的特色垂类检测模型

• 兼容PaddleDetection套件导出的其他检测模型（单类别）

• 预训练特征提取模型基于Paddle官方模型库中的Metric Learning模型开发

• 目前仅支持单类别多目标追踪

效果展示

• 行人多目标追踪：

快速使用

• 同步代码：

shell

git clone https://github.com/Wei-JL/deep_sort_paddle

• 下载预训练模型（行人多目标追踪）：链接

• 预测推理：

$ cd deep_sort_paddle

$ python main.py \
    --video_path PATH_TO_VIDEO \
    --save_dir PATH_SAVE_DIR \
    --det_model_dir DET_MODEL_DIR \
    --emb_model_dir EMB_MODEL_DIR \
    --use_gpu SET_IT_IF_USE_GPU \
    --display SET_IT_IF_DISPLAY_RESULTS