4 个月前

ReMix:在数据混合上训练通用的人再识别模型

ReMix:在数据混合上训练通用的人再识别模型

摘要

现代人物再识别(Re-ID)方法的泛化能力较弱,在捕获环境变化时会出现显著的精度下降。这是由于现有的多摄像头Re-ID数据集在规模和多样性上存在局限,因为此类数据难以获取。与此同时,有大量的未标记单摄像头记录可用。这些数据易于收集,因此更具多样性。目前,单摄像头数据仅用于Re-ID方法的自监督预训练。然而,在预训练之后通过有限的多摄像头数据进行微调会抑制单摄像头数据的多样性。在本文中,我们提出了一种名为ReMix的广义Re-ID方法,该方法联合训练了有限的标记多摄像头数据和大量的未标记单摄像头数据。通过一种新颖的数据采样策略和适应于两种类型数据联合使用的新型损失函数,我们的方法实现了有效的训练。实验结果表明,ReMix具有较高的泛化能力,并在可泛化的人物再识别任务中优于现有最先进方法。据我们所知,这是第一项探索在人物再识别中联合训练多摄像头和单摄像头数据混合体的工作。

基准测试

基准方法指标
generalizable-person-re-identification-on-21ReMix
DukeMTMC-reID-u003eRank1: 71.3
DukeMTMC-reID-u003emAP: 43.0
MSMT17-u003eRank-1: 78.2
MSMT17-u003emAP: 52.4
MSMT17-All-u003eRank-1: 84.0
MSMT17-All-u003emAP: 61.0
RandPerson-u003eRank-1: 72.7
RandPerson-u003emAP: 45.4
generalizable-person-re-identification-on-22ReMix
MSMT17-u003eRank-1: 27.3
MSMT17-u003emAP: 27.4
MSMT17-All-u003eRank-1: 37.7
MSMT17-All-u003emAP: 37.2
RandPerson-u003eRank-1: 19.3
RandPerson-u003emAP: 18.4
generalizable-person-re-identification-on-23ReMix
MSMT17-u003eRank1: 71.6
MSMT17-u003emAP: 52.8
MSMT17-All-u003eRank-1: 77.6
MSMT17-All-u003emAP: 61.6
Market-1501-u003eRank1: 58.4
Market-1501-u003emAP: 38.8
RandPerson-u003eRank1: 63.2
RandPerson-u003emAP: 42.8
person-re-identification-on-dukemtmc-reidReMix
Rank-1: 89.6
mAP: 79.8
person-re-identification-on-market-1501ReMix
Rank-1: 96.2
mAP: 89.8
person-re-identification-on-msmt17ReMix
Rank-1: 84.8
mAP: 63.9

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
ReMix:在数据混合上训练通用的人再识别模型 | 论文 | HyperAI超神经