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++ 구현을 제공합니다.
- 폭넓은 탐지 지원: 모든 모델의 축 정렬 및 회전 바운딩 박스 탐지를 지원합니다.
- 벤치마크 중심: 캐시된 벤치마크 평가 및 연구 루프를 위한 도구를 포함합니다.