基于二阶混合注意力的行人重识别
DOI:
CSTR:
作者:
作者单位:

(哈尔滨理工大学 计算机科学与技术学院,黑龙江 哈尔滨 150080)

作者简介:

崔鹏 (1971-),男,副教授,博士,主要从事图像处理和机器学习方面的研究.

通讯作者:

中图分类号:

基金项目:

黑龙江省自然科学基金(F2015038)和黑龙江教育厅(11551086)资助项目 (哈尔滨理工大学计算机科学与技术学院,黑龙江哈尔滨 150080)


Person re-identification based on second-order mixed attention
Author:
Affiliation:

(School of Computer Science and Technology,Harbin University of Science and Technology,Harbin,Heilongjiang 150080,China)

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    基于注意力机制的行人重识别方法更多利用图像中 一阶信息,忽略了特征中二阶信息 ,不能挖掘特征图之间的相关性和细粒度信息。提出一种基于二阶混合注意力的行人重 识别算法(second-order mixed attention module,SMAN)。二阶混合注意力模块(second-order mixed attention module,SOMA)由二阶通道注意力(second-order channel attention,SOCA)和二阶空间注意力模 块(second-order information,SOSA)组成,该方法将全局协方差池函数嵌入到SOCA和SOSA模块中,学习特征中二阶信息 。SOCA模块学习特征图之间相关性,SOSA模块则重新为特征图分配权重,关注特征图空间域 的细粒度信息。SMAN算法在Market-1501和 DukeMTMC-ReID数据集上的首位准确率分别 为 94.3%和87.1%,mAP分别达到85.7%和74.5%,同时使用类激活图验证SOMA模块的影响 ,实验表 明SMAN算法充分利用特征图的通道域和空间域中二阶信息。算法的性能优于现有的一些基于 注意力机制行人重识别方法,甚至接近某些优秀的方法。

    Abstract:

    Person re-identification methods based on the attention mechanism make more use of the first-order information in the feature,ignoring the second-o rd er information in the feature,and the correlation and fine-grained information between feature maps cannot be mined.This paper proposes a pedestrian re-iden t ification algorithm based on the second-order mixed attention module (SMAN) .Th e second-order mixed attention module (SOMA) is composed of the second-order ch an nel attention (SOCA) and the second-order spatial attention module (SOSA).This method imbed the Global Coordination Pooling function into SOCA and SOSA modules to learn second-order information in features.The SOCA module uses second-ord er information to learn the correlation between feature maps,While the SOSA modul e assignes weights to feature maps again,focusing on the fine-grained informat i on of feature maps spatial domain.The Rank-1accuracy of SMAN algorithm on mark e t-1501and DukeMTMC-ReID datasets reached 94.30% and 87.1%,respectively,and mA P reached 85.7% and 74.53%,respectively.Finally,the class activation map is u sed to verify the effect of SOMA module.Experiments show that SMAN algorithm ma kes full use of the second-order information in the channel domain and spatial d omain of the feature map.The performance of the algorithm is better than some ex isting person re-identification methods based on attention mechanism,and even close to the performance of some state-of-the-art methods.

    参考文献
    相似文献
    引证文献
引用本文

崔鹏,马超.基于二阶混合注意力的行人重识别[J].光电子激光,2021,32(6):645~652

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2020-12-13
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2021-08-20
  • 出版日期:
文章二维码