Structure-aware person search with self-attention and online instance aggregation matching

被引:13
作者
Gao, Cunyuan [1 ,2 ]
Yao, Rui [1 ,2 ]
Zhao, Jiaqi [1 ,2 ]
Zhou, Yong [1 ,2 ]
Hu, Fuyuan [3 ]
Li, Leida [4 ]
机构
[1] China Univ Min & Technol, Sch Comp Sci & Technol, Xuzhou 221116, Jiangsu, Peoples R China
[2] Minist Educ Peoples Republ China, Mine Digitizat Engn Res Ctr, Xuzhou 221116, Jiangsu, Peoples R China
[3] Suzhou Univ Sci & Technol, Suzhou, Peoples R China
[4] China Univ Min & Technol, Sch Informat & Control Engn, Xuzhou 221116, Jiangsu, Peoples R China
关键词
Person search; Re-Identification; Pedestrian detection;
D O I
10.1016/j.neucom.2019.08.038
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper tackles the task of person search, which is a new challenging computer version task in real-world scenarios. The challenge of this task is mainly from: i) Due to viewpoints, occlusions, etc., the visual appearance of a particular person varies greatly, making re-identification difficult; ii) the bounding box where pedestrians are not available, the model needs to search for the person in the entire gallery image. The task of person search is consisted of pedestrian detection and person re-identification (re-id). Instead of completing the two tasks separately, we put the two pieces together to address these issues. We design a more suitable end-to-end framework for person search, which both improve re-id and detection at the same time. Through the change of the anchor with structural prior, the pedestrian detection can be refined faster and better. By introducing self-attention, the framework enhances the fusion of global information. An Online Instance Aggregation Matching (OIAM) loss function is proposed to train the network, and it effectively solves the problem of many categories but few samples of the same category. Extensive experiments are conducted on the PRW and CUHK-SYSU datasets, and our proposed method can outperform other person search methods in both mAP and top-1 evaluation protocols. (C) 2019 Elsevier B.V. All rights reserved.
引用
收藏
页码:29 / 38
页数:10
相关论文
共 49 条
[1]  
[Anonymous], 2015, PROC CVPR IEEE, DOI 10.1109/CVPR.2015.7299016
[2]   A survey of approaches and trends in person re-identification [J].
Bedagkar-Gala, Apurva ;
Shah, Shishir K. .
IMAGE AND VISION COMPUTING, 2014, 32 (04) :270-286
[3]   A non-local algorithm for image denoising [J].
Buades, A ;
Coll, B ;
Morel, JM .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 2, PROCEEDINGS, 2005, :60-65
[4]   Learning Complexity-Aware Cascades for Deep Pedestrian Detection [J].
Cai, Zhaowei ;
Saberian, Mohammad ;
Vasconcelos, Nuno .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :3361-3369
[5]  
Chang X., 2018, MULTI LEVEL FACTORIS
[6]   RCAA: Relational Context-Aware Agents for Person Search [J].
Chang, Xiaojun ;
Huang, Po-Yao ;
Shen, Yi-Dong ;
Liang, Xiaodan ;
Yang, Yi ;
Hauptmann, Alexander G. .
COMPUTER VISION - ECCV 2018, PT IX, 2018, 11213 :86-102
[7]   Histograms of oriented gradients for human detection [J].
Dalal, N ;
Triggs, B .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, :886-893
[8]   Deep feature learning with relative distance comparison for person re-identification [J].
Ding, Shengyong ;
Lin, Liang ;
Wang, Guangrun ;
Chao, Hongyang .
PATTERN RECOGNITION, 2015, 48 (10) :2993-3003
[9]   Fast Feature Pyramids for Object Detection [J].
Dollar, Piotr ;
Appel, Ron ;
Belongie, Serge ;
Perona, Pietro .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2014, 36 (08) :1532-1545
[10]   Person Re-Identification by Symmetry-Driven Accumulation of Local Features [J].
Farenzena, M. ;
Bazzani, L. ;
Perina, A. ;
Murino, V. ;
Cristani, M. .
2010 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2010, :2360-2367