An Improved One-Stage Pedestrian Detection Method Based on Multi-Scale Attention Feature Extraction

Ma Jun,Wan Honglin,Wang Junxia,Xia Hao,Bai Chengjie
DOI: https://doi.org/10.1007/s11554-021-01074-2
IF: 2.293
2021-01-01
Journal of Real-Time Image Processing
Abstract:In recent years, the performance of the convolutional neural network-based pedestrian detection method has improved significantly. However, an imbalance remains between detection accuracy and speed. In this paper, we employ a one-stage object detection framework and propose a pedestrian detection method based on the multi-scale attention mechanism of a convolutional neural network to improve the imbalance between accuracy and speed. First, a multi-scale convolution module is designed to extract corresponding features at different scales. Second, using the attention module, association information between features is mined from space and channel perspectives to strengthen the original features. Then, the enhanced features are passed through a classification and regression module to perform object positioning and bounding box regression. Finally, to learn more pedestrian location information, we improve the loss function to realise better network training. The proposed method achieved considerable results on the challenging CityPersons and Caltech pedestrian detection datasets.
What problem does this paper attempt to address?