Multi-Object Tracking based on Optimal Transport and Coordinate Attention Mechanism

Multi-Object Tracking (MOT) has currently attracted significant interest due to its wide applications in various fields, such as autonomous driving, intelligent surveillance, and behavior recognition. However, appearance similarity of different objects results in low accuracy of target matching and difficulties in data association. In this paper, we propose a Multi-Object Tracking based on Optimal Transport and Coordinate Attention Mechanism (MOT2A), which addresses above challenges by integrating the attention mechanism with optimal transport. These strategies effectively enhance the extraction of discriminative appearance features and improve target matching between different frames. Firstly, we construct a novel Coordinate attention module (CASA), which models the interdependence between the channel domain and the spatial domain of the feature map. Secondly, a Triplet loss with optimal transport (SK-Triplet) is designed to adjust the distance matrix for effective clustering of positive and negative samples during loss calculation. Finally, extensive experiments are conducted on MOT17 and MOT20. For MOT17: 79.4 MOTA, 78.9 IDF1, and 63.9 HOTA; For MOT20: 77.0 MOTA, 76.3 IDF1, and 62.3 HOTA are achieved, respectively. Compared to existing MOT methods, our method shows significant improvements in accuracy and stability. The code is available at: https://github.com/420-s/MOT2A.