HyperAIHyperAI

Command Palette

Search for a command to run...

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

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)上进行了实验验证,结果表明,在五个广泛使用的公开数据集上均取得了显著的性能提升。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供