将FairMOT的多目标跟踪使用的卡尔曼滤波器改为cpp实现, 输入输出效果与原来的一致,提升了追踪算法运行的速度。
项目使用pytorch extension实现,不过运算没有用到torch矩阵。
使用的是Eigen库,因此需要安装pytorch以及Eigen。
Eigen安装:sudo apt install libeigen3-dev
cd tracklet
conda activate ${your_env}
python setup.py install
将会安装一个名为tracklet的包。调用的话将原有的调用kalman_filter.py替换成如下即可。
# from tracking_utils.kalman_filter import KalmanFilter
from tracklet.kalman_filter import KalmanFilter
原使用demo.py对示例视频追踪,使用yolo检测器,推理20ms,追踪算法运行10ms。 修改后,追踪结果不变, 追踪算法运行5ms。
目前只修改卡尔曼滤波器部分,下一步进行更上层的修改。