A Multi Object Tracking Framework Based on YOLOv8s and Bytetrack Algorithm

Autor: Yingyun Wang, Vladimir Y. Mariano
Jazyk: angličtina
Rok vydání: 2024
Předmět:
Zdroj: IEEE Access, Vol 12, Pp 120711-120719 (2024)
Druh dokumentu: article
ISSN: 2169-3536
DOI: 10.1109/ACCESS.2024.3450370
Popis: In recent years, the YOLOv8 series algorithms have become a research hotspot in many fields, and they can perform excellently in different computer vision tasks. However, YOLOv8 still has room for improvement in multi-target tracking. We integrated it with the Symmetric Positive Definite Convolution (SPD-Conv) module and proposed the YOLOv8s SPD detector, which enhances its detection ability for small targets. The values of mAP@0.5 and mAP@.5:95 have both been increased compared to YOLOv8s. Subsequently, the detector was combined with the ByteTrack tracking algorithm, and the IoU and loss function were optimized to achieve superior performance. We refer to this tracking framework as YBTrack. YBTrack was tested on the Multiple Object Tracking (MOT) Challenge MOT17 and MOT 20 datasets, and achieved MOTA metrics of 74.0% and 66.8%, respectively. Compared with existing tracking frameworks with built-in detectors, our tracking framework has better performance.
Databáze: Directory of Open Access Journals