Pedestrian attribute recognition based on dual self-attention Mechanism
- School of Communication and Information Engineering, Shanghai University,
200444 shanghai, China
{fanzkui, ypguan}@shu.edu.cn - Key Laboratory of Advanced Displays and System Application, Ministry of Education,
200444 shanghai, China
Abstract
Recognizing pedestrian attributes has recently obtained increasing attention due to its great potential in person re-identification, recommendation system, and other applications. Existing methods have achieved good results, but these methods do not fully utilize region information and the correlation between attributes. This paper aims at proposing a robust pedestrian attribute recognition framework. Specifically, we first propose an end-to-end framework for attribute recognition. Secondly, spatial and semantic self-attention mechanism is used for key points localization and bounding boxes generation. Finally, a hierarchical recognition strategy is proposed, the whole region is used for the global attribute recognition, and the relevant regions are used for the local attribute recognition. Experimental results on two pedestrian attribute datasets PETA and RAP show that the mean recognition accuracy reaches 84.63% and 82.70%. The heatmap analysis shows that our method can effectively improve the spatial and the semantic correlation between attributes. Compared with existing methods, it can achieve better recognition effect.
Key words
pedestrian attribute recognition; spatial self-attention; semantic self-attention; deep learning
Digital Object Identifier (DOI)
https://doi.org/10.2298/CSIS220815016F
Publication information
Volume 20, Issue 2 (April 2023)
Special Issue on Machine Learning-based Decision Support Systems in IoT systems
Year of Publication: 2023
ISSN: 2406-1018 (Online)
Publisher: ComSIS Consortium
Full text
Available in PDF
Portable Document Format
How to cite
Fan, Z., Guan, Y.: Pedestrian attribute recognition based on dual self-attention Mechanism. Computer Science and Information Systems, Vol. 20, No. 2, 793–812. (2023), https://doi.org/10.2298/CSIS220815016F