全局注意力机制与关系网络驱动的行人重识别

作     者：刘慧 梁东升 张雷 卢云志 LIU Hui;LIANG Dongsheng;ZHANG Lei;LU Yunzhi

作者机构：北京建筑大学电气与信息工程学院北京100044 

基　　金：国家自然科学基金资助项目(61501019) 

出 版 物：《中国科技论文》 (China Sciencepaper)

年 卷 期：2023年第18卷第7期

页      码：759-765,785页

摘      要：针对行人重识别(re-identification,ReID)中因遮挡、姿态变化导致模型提取关键特征能力不足的问题,提出一种全局注意力机制与关系网络驱动的行人重识别算法。首先,在主干网络部分,将全局注意力模块嵌入ResNet-50网络中,捕获空间维度和通道维度的权重信息;其次,在关系网络部分,通过水平切分得到不同尺度的多个局部特征,并设计了2个模块分别用于提取全局对比特征及局部特征间的关系特征,并将局部关系特征与全局对比信息融合后送到分类网络;最后,在损失优化阶段,采用Circle Loss、三元组损失与交叉熵损失的联合损失进行网络训练。在5个常用数据集上进行实验并对实验结果进行分析,其中Occluded-DukeMTMC和Market1501数据集上rank-1值分别达到63.2%和95.4%,mAP(mean average precision)值分别为53.8%和88.2%,体现了所提算法的先进性。

主 题 词：行人重识别 注意力机制 关系网络 多尺度 联合损失 

学科分类：1305[艺术学-设计学类] 13[艺术学] 08[工学] 081104[081104] 0804[工学-材料学] 081101[081101] 0811[工学-水利类] 

D　O　I：10.3969/j.issn.2095-2783.2023.07.010

馆 藏 号：203122674...