3 个月前

基于场景解耦学习的人体重识别

基于场景解耦学习的人体重识别

摘要

行人重识别(ReID)任务面临诸多挑战,例如遮挡和尺度变化等问题。现有方法通常采用单分支网络来应对这些挑战,但这种网络需要具备对各种复杂情况的强鲁棒性,导致网络负担过重。本文提出一种“分而治之”的策略来解决ReID任务。为此,我们引入多种自监督操作,模拟不同的挑战性场景,并通过不同的子网络分别处理各类问题。具体而言,我们采用随机擦除(random erasing)操作,并提出一种新颖的随机缩放(random scaling)操作,以生成具有可控特性的新图像。为此,我们设计了一种通用的多分支网络结构,包含一个主分支和两个从属分支,用于应对不同场景。各分支之间协同学习,从而获得差异化的感知能力。通过该机制,ReID任务中复杂的场景被有效解耦,各分支的计算负担也得以减轻。大量实验结果表明,所提出方法在三个标准ReID基准数据集以及两个遮挡ReID基准数据集上均取得了当前最优的性能表现。消融实验进一步验证了所提出的网络结构与自监督操作在多种场景下均能显著提升模型性能。代码已开源,地址为:https://git.openi.org.cn/zangxh/LDS.git。

代码仓库

deropty/LDS
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
person-re-identification-on-dukemtmc-reidLDS (ResNet50 + RK)
Rank-1: 92.91
mAP: 91.0
person-re-identification-on-market-1501LDS (ResNet50 + RK)
Rank-1: 96.17
mAP: 94.89
person-re-identification-on-msmt17LDS (ResNet50+RK)
Rank-1: 88.35
mAP: 79.09
person-re-identification-on-occluded-dukemtmcLDS
Rank-1: 64.39
mAP: 55.75
person-re-identification-on-p-dukemtmc-reidLDS
Rank-1: 91.96
Rank-10: 96.39
Rank-5: 95.28
mAP: 82.93

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于场景解耦学习的人体重识别 | 论文 | HyperAI超神经