
摘要
域适应在行人重识别(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-to | SSG | mAP: 58.3 rank-1: 80.0 rank-10: 92.4 rank-5: 90.0 |
| unsupervised-domain-adaptation-on-duke-to-1 | SSG | mAP: 13.3 rank-1: 32.2 rank-10: 51.2 rank-5: - |
| unsupervised-domain-adaptation-on-market-to | SSG | mAP: 53.4 rank-1: 73.0 rank-10: 83.2 rank-5: 80.6 |
| unsupervised-domain-adaptation-on-market-to-1 | SSG | mAP: 13.2 rank-1: 31.6 rank-10: 49.6 rank-5: - |
| unsupervised-person-re-identification-on-2 | Self-Similarity Grouping (one shot) | Rank-1: 27.6 Rank-10: 45.7 mAP: 11.8 |
| unsupervised-person-re-identification-on-3 | Self-Similarity Grouping (one shot) | Rank-1: 43.6 Rank-10: 61.8 mAP: 23.6 |
| unsupervised-person-re-identification-on-4 | Self-Similarity Grouping (one shot) | MAP: 71.5 Rank-1: 87.5 Rank-10: 96.8 Rank-5: 95.2 |
| unsupervised-person-re-identification-on-5 | Self-Similarity Grouping (one shot) | MAP: 55.9 Rank-1: 72.4 Rank-10: 87.7 Rank-5: 84 |