Dynamic Weighting Network for Person Re-Identification

被引:2
作者
Li, Guang [1 ,2 ]
Liu, Peng [2 ,3 ]
Cao, Xiaofan [1 ,2 ]
Liu, Chunguang [2 ]
机构
[1] North China Elect Power Univ, Sch Control & Comp Engn, Beijing 102206, Peoples R China
[2] North China Elect Power Univ, Yangzhong Intelligent Elect Res Ctr, Yangzhong 212211, Peoples R China
[3] Changchun Univ Sci & Technol, Sch Elect & Informat Engn, Changchun 130012, Peoples R China
关键词
re-identification; self-attention; fine-grained features;
D O I
10.3390/s23125579
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
Recently, hybrid Convolution-Transformer architectures have become popular due to their ability to capture both local and global image features and the advantage of lower computational cost over pure Transformer models. However, directly embedding a Transformer can result in the loss of convolution-based features, particularly fine-grained features. Therefore, using these architectures as the backbone of a re-identification task is not an effective approach. To address this challenge, we propose a feature fusion gate unit that dynamically adjusts the ratio of local and global features. The feature fusion gate unit fuses the convolution and self-attentive branches of the network with dynamic parameters based on the input information. This unit can be integrated into different layers or multiple residual blocks, which will have varying effects on the accuracy of the model. Using feature fusion gate units, we propose a simple and portable model called the dynamic weighting network or DWNet, which supports two backbones, ResNet and OSNet, called DWNet-R and DWNet-O, respectively. DWNet significantly improves re-identification performance over the original baseline, while maintaining reasonable computational consumption and number of parameters. Finally, our DWNet-R achieves an mAP of 87.53%, 79.18%, 50.03%, on the Market1501, DukeMTMC-reID, and MSMT17 datasets. Our DWNet-O achieves an mAP of 86.83%, 78.68%, 55.66%, on the Market1501, DukeMTMC-reID, and MSMT17 datasets.
引用
收藏
页数:15
相关论文
共 37 条
  • [1] Ba J.L., 2016, arXiv
  • [2] Attention Augmented Convolutional Networks
    Bello, Irwan
    Zoph, Barret
    Vaswani, Ashish
    Shlens, Jonathon
    Le, Quoc V.
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 3285 - 3294
  • [3] Self-Critical Attention Learning for Person Re-Identification
    Chen, Guangyi
    Lin, Chunze
    Ren, Liangliang
    Lu, Jiwen
    Zhou, Jie
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 9636 - 9645
  • [4] Batch DropBlock Network for Person Re-identification and Beyond
    Dai, Zuozhuo
    Chen, Mingqiang
    Gu, Xiaodong
    Zhu, Siyu
    Tan, Ping
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 3690 - 3700
  • [5] Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929
  • [6] CMT: Convolutional Neural Networks Meet Vision Transformers
    Guo, Jianyuan
    Han, Kai
    Wu, Han
    Tang, Yehui
    Chen, Xinghao
    Wang, Yunhe
    Xu, Chang
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, : 12165 - 12175
  • [7] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [8] He LX, 2020, Arxiv, DOI arXiv:2006.02631
  • [9] TransReID: Transformer-based Object Re-Identification
    He, Shuting
    Luo, Hao
    Wang, Pichao
    Wang, Fan
    Li, Hao
    Jiang, Wei
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 14993 - 15002
  • [10] Interaction-and-Aggregation Network for Person Re-identification
    Hou, Ruibing
    Ma, Bingpeng
    Chang, Hong
    Gu, Xinqian
    Shan, Shiguang
    Chen, Xilin
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 9309 - 9318