학술논문

Lightweight high-precision pedestrian tracking algorithm in complex occlusion scenarios
Document Type
Article
Source
KSII Transactions on Internet and Information Systems (TIIS). Jan 01, 2023 17(3):840
Subject
object tracking
attention mechanism
object detection
non-maximum suppression
lightweight neural network
Language
English
ISSN
1976-7277
Abstract
Aiming at the serious occlusion and slow tracking speed in pedestrian target tracking and recognition in complex scenes, a target tracking method based on improved YOLO v5 combined with Deep SORT is proposed. By merging the attention mechanism ECA-Net with the Neck part of the YOLO v5 network, using the CIoU loss function and the method of CIoU non-maximum value suppression, connecting the Deep SORT model using Shuffle Net V2 as the appearance feature extraction network to achieve lightweight and fast speed tracking and the purpose of improving tracking under occlusion. A large number of experiments show that the improved YOLO v5 increases the average precision by 1.3% compared with other algorithms. The improved tracking model, MOTA reaches 54.3% on the MOT17 pedestrian tracking data, and the tracking accuracy is 3.7% higher than the related algorithms and The model presented in this paper improves the FPS by nearly 5 on the fps indicator.