3 个月前

相机感知代理用于无监督行人重识别

相机感知代理用于无监督行人重识别

摘要

本文针对完全无监督的人体重识别(Re-ID)问题,该问题无需任何标注信息。以往一些方法采用聚类技术生成伪标签,并利用生成的标签逐步训练Re-ID模型,这类方法相对简单但效果良好。然而,大多数基于聚类的方法将每个聚类视为一个伪身份类别,忽略了由于摄像头视角变化所导致的较大类内差异(intra-ID variance)。为解决这一问题,本文提出将每个单一聚类进一步划分为多个代理(proxy),每个代理代表来自同一摄像头的样本实例。这些具有摄像头感知能力的代理能够有效应对较大的类内差异,并生成更为可靠的伪标签以支持模型学习。基于上述摄像头感知代理,本文设计了针对相机内(intra-camera)与跨相机(inter-camera)的对比学习组件,以有效提升模型在单个摄像头及跨摄像头场景下的身份判别能力。同时,还提出一种代理均衡采样策略,进一步促进模型的训练效果。在三个大规模Re-ID数据集上的大量实验表明,所提出的方法显著优于大多数现有的无监督方法。尤其在具有挑战性的MSMT17数据集上,相较于第二名,本方法在Rank-1准确率上提升了14.3%,mAP提升了10.2%。代码已开源,地址为:\texttt{https://github.com/Terminator8758/CAP-master}。

代码仓库

Terminator8758/CAP-master
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
person-re-identification-on-dukemtmc-reidCAP
Rank-1: 87.7
Rank-10: 95.4
Rank-5: 93.7
mAP: 76
person-re-identification-on-market-1501CAP
Rank-1: 93.3
Rank-5: 97.5
mAP: 85.1
unsupervised-person-re-identification-on-4CAP
MAP: 79.2
Rank-1: 91.4
Rank-10: 97.7
Rank-5: 96.3
unsupervised-person-re-identification-on-5CAP
MAP: 67.3
Rank-1: 81.1
Rank-10: 91.8
Rank-5: 89.3

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
相机感知代理用于无监督行人重识别 | 论文 | HyperAI超神经