计算机工程与应用››2024,第60卷››问题(4): 133-141.内政部:10.3778/j.issn.1002-8331.2209-0120

模式识别与人工智能 • 上一篇   下一篇

结合数据增强与特征融合的跨模态行人重识别

宋雨,王帮海,曹钢钢  

  1. 广东工业大学  计算机学院,广州  510006
  • 出版日期:2024-02-15 发布日期:2024-02-15

结合数据增强和特征融合的跨模态人识别

宋瑜、王邦海、曹刚刚  

  1. 广东工业大学计算机科学学院,广州510006
  • 在线:2024-02-15 出版:2024-02-15

摘要:可见光-红外行人重识别问题的难点在于图像间模态差异大,大多数现有的方法通过生成对抗网络生成伪图像或提取原始图像上的模态共享特征来缓解模态差异。然而,训练生成对抗网络需要消耗大量的计算资源且生成的伪图像容易引入噪声,提取模态共享特征也会不可避免地导致与行人身份相关的重要判别特征丢失。针对以上问题,提出新的跨模态行人重识别网络。首先将进行自动数据增强后的训练数据集作为网络输入,提高模型的鲁棒性;然后在网络中引入实例正则化来缩小模态差异;最后将网络各层提取到的不同尺度的行人特征进行有机融合,融合后的特征包含更多与行人身份相关的判别特征。该方法在SYSU-MM01系统数据集的全局搜索模式下排名-1/mAP,69.47%/65.05%,RegDB数据集的可见光到红外模式下排名-1/mAP回报率85.73%/77.77%实验结果获得显著提升。

关键词: 跨模态, 行人重识别, 自动数据增强, 特征融合

摘要:可见红外人物再识别问题的难点在于图像之间存在较大的模态差异。现有的方法大多通过生成对抗网络生成伪图像或在原始图像上提取模态共享特征来缓解模态差异。然而,生成性对抗网络的训练消耗了大量的计算资源,生成了容易引入噪声的伪图像,提取模式共享特征也会导致重要的区分特征丢失。为了解决这些问题,提出了一种新的跨模态人员再识别网络。首先,使用自动数据增强来提高模型的鲁棒性。然后,在网络中使用实例正则化来减少模态差异。最后,对网络各层提取的不同尺度的行人特征进行有机融合,融合后的特征包含更多与行人身份相关的差异化特征。该方法在SYSU-MM01全搜索模式下的秩-1/mAP为69.47%/65.05%,在RegDB可见-红外模式下的Rank-1/mAP为85.73%/77.77%。实验结果有显著的改善效果。

关键词: 交叉模态, 人员重新识别, 自动数据增强, 特征融合