4 个月前

自相似性聚类:一种简单的人再识别无监督跨域适应方法

自相似性聚类:一种简单的人再识别无监督跨域适应方法

摘要

域适应在行人重识别(re-ID)中一直是一个具有挑战性的任务。在这项工作中,我们探讨了如何利用目标域样本中存在的自然相似特性,以无监督的方式进行行人重识别的学习。具体而言,我们提出了一种自相似分组(Self-similarity Grouping, SSG)方法,该方法利用未标记样本的潜在相似性(从整体身体到局部部位)自动从不同视角构建多个聚类。这些独立的聚类随后被分配标签,作为伪身份来监督训练过程。我们反复交替执行这一分组和训练过程,直到模型稳定。尽管方法看似简单,但我们的SSG在mAP指标上分别超越了现有最佳方法4.6%(DukeMTMC到Market1501)和4.4%(Market1501到DukeMTMC)。基于我们的SSG方法,我们进一步引入了一种名为SSG++的聚类引导半监督方法,在开放集设置下进行单次域适应(即目标域中的独立身份数量未知)。无需花费大量精力进行标注,我们的SSG++方法可以进一步将mAP指标提升10.7%和6.9%。我们的代码可在以下地址获取:https://github.com/OasisYang/SSG 。

代码仓库

OasisYang/SSG
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
unsupervised-domain-adaptation-on-duke-toSSG
mAP: 58.3
rank-1: 80.0
rank-10: 92.4
rank-5: 90.0
unsupervised-domain-adaptation-on-duke-to-1SSG
mAP: 13.3
rank-1: 32.2
rank-10: 51.2
rank-5: -
unsupervised-domain-adaptation-on-market-toSSG
mAP: 53.4
rank-1: 73.0
rank-10: 83.2
rank-5: 80.6
unsupervised-domain-adaptation-on-market-to-1SSG
mAP: 13.2
rank-1: 31.6
rank-10: 49.6
rank-5: -
unsupervised-person-re-identification-on-2Self-Similarity Grouping (one shot)
Rank-1: 27.6
Rank-10: 45.7
mAP: 11.8
unsupervised-person-re-identification-on-3Self-Similarity Grouping (one shot)
Rank-1: 43.6
Rank-10: 61.8
mAP: 23.6
unsupervised-person-re-identification-on-4Self-Similarity Grouping (one shot)
MAP: 71.5
Rank-1: 87.5
Rank-10: 96.8
Rank-5: 95.2
unsupervised-person-re-identification-on-5Self-Similarity Grouping (one shot)
MAP: 55.9
Rank-1: 72.4
Rank-10: 87.7
Rank-5: 84

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
自相似性聚类:一种简单的人再识别无监督跨域适应方法 | 论文 | HyperAI超神经