Pedestrian Tracking Algorithm Combining Contextual Information and Attention Mechanism

Abstract

In the real scene, because pedestrians are occluded or the size of pedestrians is small, the convolutional neural network cannot fully extract their features, resulting in poor detection results. In two adjacent frames, the same pedestrian is prone to errors when doing data association, which makes the pedestrian tracking effect unsatisfactory. In order to solve this problem, the pedestrian tracking algorithm based on Anchor-free idea is improved. A fusion context information module is proposed to enhance the model's feature extraction ability for different receptive fields, and improve the model's detection and tracking performance when the pedestrian size is small. In addition, in order to let the model learn to pay attention to the effective information of the feature layer. A coordinated attention mechanism is introduced to guide the model to learn the weights of different channels and different regions of the feature layer, and to improve the tracking performance of the model when pedestrians are occluded. In the experiment, the tracking performance of the model was verified on the MOT16 dataset. Experimental results show that compared with other main popular person tracking algorithms, the improved algorithm has higher tracking accuracy and lower pedestrian ID switching times. Its tracking accuracy is 70.74.