中文

基于度量学习构建计算高效且泛化能力强的行人重识别模型

计算机视觉与模式识别 2021-08-19 v2

摘要

本文研究行人重识别中的域偏移问题。在一个数据集上训练的模型,在未见数据上通常表现差得多。这一差距部分源于行人重识别数据集规模相对较小(例如与人脸识别数据集相比),但也与训练目标有关。我们提出使用度量学习目标,即 AM-Softmax 损失,以及一些额外的训练实践,来构建泛化能力强且计算高效的模型。我们采用近期提出的 Omni-Scale Network (OSNet) 架构,结合若干训练技巧与架构调整,在大规模 MSMT17 数据集的三种设定下(MSMT17-all->DukeMTMC、MSMT17-train->Market1501 和 MSMT17-all->Market1501)取得了跨域泛化问题的当前最优结果。

关键词

引用

@article{arxiv.2003.07618,
  title  = {Building Computationally Efficient and Well-Generalizing Person Re-Identification Models with Metric Learning},
  author = {Vladislav Sovrasov and Dmitry Sidnev},
  journal= {arXiv preprint arXiv:2003.07618},
  year   = {2021}
}

备注

Submitted to International Conference on Pattern Recognition (ICPR 2020)