mikel-brostrom/boxmot
BoxMOT: Pluggable Python and C++ SOTA multi-object tracking modules with support for axis-aligned and oriented bounding boxes
解决的问题
BoxMOT 为多目标跟踪 (MOT) 工作流提供统一的接口。它消除了为每个实验重新构建整个检测器和跟踪器堆栈的需要,允许用户使用单个 CLI 和 Python API 轻松更换跟踪器并针对基准测试进行评估。
工作原理
该项目实现了一种可插拔架构,可以在共享相同检测器和 ReID(重识别)组件的同时更换跟踪器。它支持轴对齐边界框 (AABB) 和有向边界框 (OBB),并提供用于生产环境的 Python 和原生 C++ 跟踪器实现。
适用对象
专为从事计算机视觉任务的研究人员和开发人员设计,这些任务需要跨视频帧跟踪多个对象,特别是那些需要用于调优和评估的基准测试导向工作流的用户。
亮点
- 统一接口:用于跟踪、生成、评估、调优和训练 ReID 模型的单一 CLI 和 API。
- 可更换跟踪器:支持 OccluBoost、BoTSORT、ByteTrack 等多种现代跟踪器。
- 双后端:提供用于灵活性的 Python 实现和用于性能的生产级原生 C++ 实现。
- 广泛的检测支持:适用于来自任何模型的轴对齐和有向边界框检测。
- 面向基准测试:包含用于缓存基准测试评估和研究循环的工具。