Learning deep part-aware embedding for person retrieval

Yang Zhao, Chunhua Shen, Xiaohan Yu*, Hao Chen, Yongsheng Gao, Shengwu Xiong

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

25 Citations (Scopus)

Abstract

Person retrieval is an important vision task, aiming at matching the images of the same person under various camera views. The key challenge of person retrieval lies in the large intra-class variations among the person images. Therefore, how to learn discriminative feature representations becomes the core problem. In this paper, we propose a deep part-aware representation learning method for person retrieval. First, an improved triplet loss is introduced such that the global feature representations from the same identity are closely clustered. Meanwhile, a localization branch is proposed to automatically localize those discriminative person-wise parts or regions, only using identity labels in a weakly supervised manner. Via the learning simultaneously guided by the global branch and the localization branch, the proposed method can further improve the performance for person retrieval. Through an extensive set of ablation studies, we verify that the localization branch and the improved triplet loss each contributes to the performance boosts of the proposed method. Our model obtains superior (or comparable) performance compared to state-of-the-art methods for person retrieval on the four public person retrieval datasets. On the CUHK03-labeled dataset, for instance, the performance increases from 73.0% mAP and 77.9% rank-1 accuracy to 80.8% (+7.8%) mAP and 83.9% (+6.0%) rank-1 accuracy.

Original languageEnglish
Article number107938
Pages (from-to)1-10
Number of pages10
JournalPattern Recognition
Volume116
DOIs
Publication statusPublished - Aug 2021
Externally publishedYes

Keywords

  • Person retrieval
  • Part-aware embedding
  • Improved triplet loss

Fingerprint

Dive into the research topics of 'Learning deep part-aware embedding for person retrieval'. Together they form a unique fingerprint.

Cite this