# PaddleDeepSort

**Repository Path**: oscarlin/paddle-deep-sort

## Basic Information

- **Project Name**: PaddleDeepSort
- **Description**: 基于飞桨实现的deepsort算法
- **Primary Language**: Python
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 3
- **Forks**: 0
- **Created**: 2021-05-23
- **Last Updated**: 2025-01-31

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# PaddleDeepSort

#### 介绍
基于PaddleDetection目标检测库的DeepSort算法，支持基于PaddleDtection中各种检测模型的DeepSort
<img src="out (1) 00_00_00-00_00_30.gif" alt="show" />


#### 快速开始[paddle必须在GPU上运行]
##### 方式一
1.获取本仓库
```python
git clone https://gitee.com/oscarlin/paddle-deep-sort.git
cd paddle-deep-sort
```
1.安装Paddle以及PaddleDetection

[具体参考教程](https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.1/docs/tutorials/INSTALL_cn.md)
注意：该代码支持paddle2.1版本，若为2.0.版本，utils.py 66:行报错修改指南： paddle.float32->'float32'

2.将deepsort文件夹复制到PaddleDetection文件夹内

```
eg:PaddleDetection/deepsort
```

3.将deepsort文件夹内的infer.py 和infer_from_bag.py, utils.py文件复制到PaddleDetection文件夹下

```
eg:PaddleDetection/infer_from_bag.py
```

##### 方式二
我们提供AI Studio上的[演示项目](https://aistudio.baidu.com/aistudio/projectdetail/1985522?channel=0&channelType=0&shared=1)，直接可以在线运行，欢迎fork


#### 开始推理
注意：生成的视频路径必须为mp4格式
##### 1.视频推理
```python
python infer.py -video [视频路径] --path [生成视频路径] -config [检测模型配置文件]\
 -o [检测模型参数路径]
```
例如：
```python
python infer.py -video demo/IMG_2931.MP4 --path out1.mp4 -config configs/pedestrian/pedestrian_yolov3_darknet.yml\
 -o https://paddledet.bj.bcebos.com/models/pedestrian_yolov3_darknet.pdparams
```

##### 2.从bag文件读取
该部分依赖rosbag库，需要首先安装
```python
pip install --extra-index-url https://rospypi.github.io/simple/ rospy rosbag
```

```python
python infer_from_bag.py -video [bag文件路径]\
 -path [输出视频路径] -fps=[fps] -topic=[bag文件内对应topic的key] \
 -config [检测模型配置文件] \
 -o [检测模型参数路径]
```
例如：
```python
python infer_from_bag.py -video ../data/data90998/2021-05-20-22-08-08.bag\
 -path o.mp4 -fps=26 -topic=/camera/color/image_raw \
 -config configs/pedestrian/pedestrian_yolov3_darknet.yml \
 -o https://paddledet.bj.bcebos.com/models/pedestrian_yolov3_darknet.pdparams
```


#### 参考实现
1.[PaddleDetection](https://github.com/PaddlePaddle/PaddleDetection)

2.[mikel-brostrom/Yolov5_DeepSort_Pytorch](https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch)