分享最近做的一个关于在线多人追踪的项目

[xieincz]

检测器采用的是 YOLOv5 （可以轻松替换成其他同类的检测器），追踪器采用的是 ByteTrack 。和其他同类项目不同的是加入了多线程处理以及用 C++和 TensorRT 加快推理速度。一个线程负责读取来自视频文件（可以轻松更改为摄像头）的帧，一个线程负责用 YOLOv5 得到检测框，一个线程用 ByteTrack 给各个检测框 reid ，还有一个线程负责将结果绘制到视频文件中。在多 CPU 核心（>=3 核）的设备上的效果比目前其他同类的项目要更快。而且还可以根据需要给各个线程设置 CPU 亲和性（将某线程绑定到某个 CPU 核心，该功能仅限于 Linux 平台）。

此外还利用 SWIG 包装了接口，方便在 python 中像调用一个库一样使用本项目。

欢迎大家来试用 ：）

Github: YOLOv5_ByteTrack_Multithreading_TensorRT

[hifans]

@xieincz ，请问有兴趣合作没？

[LXGMAX]

玩这么大？