4 个月前

重新思考基于摄像机的批量归一化在人员再识别中的分布差异问题

重新思考基于摄像机的批量归一化在人员再识别中的分布差异问题

摘要

人员重识别(ReID)的基本难题在于学习不同摄像头之间的对应关系。这需要高昂的跨摄像头注释成本,然而训练好的模型并不能保证在以前未见过的摄像头中表现良好。这些问题极大地限制了ReID的应用。本文重新审视了传统ReID方法的工作机制,并提出了一种新的解决方案。通过一种名为基于摄像头的批归一化(Camera-based Batch Normalization, CBN)的有效算子,我们强制所有摄像头的图像数据落在同一子空间内,从而大幅缩小任意一对摄像头之间的分布差异。这种对齐带来了两个好处。首先,训练好的模型在包含未见过摄像头的不同场景中具有更好的泛化能力,同时也能在多个训练集之间实现良好的迁移。其次,我们可以依赖之前因缺乏跨摄像头信息而被低估的单个摄像头注释来实现具有竞争力的ReID性能。广泛的实验结果表明了我们方法的有效性。代码已发布在 https://github.com/automan000/Camera-based-Person-ReID。

代码仓库

automan000/Camera-based-Person-ReID
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
person-re-identification-on-dukemtmc-reidCBN
Rank-1: 82.5
Rank-10: 94.1
Rank-5: 91.7
mAP: 67.3
person-re-identification-on-dukemtmc-reidCBN+BoT*
Rank-1: 84.8
Rank-10: 95.2
Rank-5: 92.5
mAP: 70.1
person-re-identification-on-market-1501CBN+BoT*
Rank-1: 94.3
Rank-5: 97.9
mAP: 83.6
person-re-identification-on-market-1501CBN
Rank-1: 91.3
Rank-5: 97.1
mAP: 77.3
person-re-identification-on-msmt17CBN
Rank-1: 72.8
mAP: 42.9
unsupervised-domain-adaptation-on-duke-toCBN+ECN
mAP: 52
rank-1: 81.7
rank-10: 94.7
rank-5: 91.9
unsupervised-domain-adaptation-on-market-toCBN+ECN
mAP: 44.9
rank-1: 68
rank-10: 83.9
rank-5: 80

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
重新思考基于摄像机的批量归一化在人员再识别中的分布差异问题 | 论文 | HyperAI超神经