17 天前

基于深度注意力感知特征学习的人体重识别

Yifan Chen, Han Wang, Xiaolu Sun, Bin Fan, Chu Tang
基于深度注意力感知特征学习的人体重识别
摘要

视觉注意力机制已被证明能够有效提升行人重识别(person re-identification, ReID)的性能。现有大多数方法通过学习一个额外的注意力图来重新加权特征图,从而以启发式方式引入视觉注意力。然而,这类方法不可避免地增加了模型复杂度和推理时间。本文提出一种新策略:在不改变原始网络结构的前提下,将注意力学习作为附加目标融入行人ReID网络中,从而在保持原有推理时间与模型规模不变的情况下实现性能提升。本文考虑了两种类型的注意力机制,分别使学习到的特征图关注整体行人信息与局部身体部位信息。在全局层面,提出全貌注意力分支(Holistic Attention Branch, HAB),引导骨干网络提取的特征图聚焦于行人主体,以减轻背景干扰。在局部层面,引入部分注意力分支(Partial Attention Branch, PAB),将提取的特征解耦为多个子组,并分别负责不同身体部位(即关键点)的表征,从而增强对姿态变化和部分遮挡的鲁棒性。这两种注意力机制具有通用性,可无缝集成至现有的各类ReID网络中。我们在两种典型网络(TriNet 和 Bag of Tricks)上进行了实验验证,结果表明,在五个广泛使用的公开数据集上均取得了显著的性能提升。