4 个月前

由GAN生成的未标记样本在体外改善了行人重识别基线模型

由GAN生成的未标记样本在体外改善了行人重识别基线模型

摘要

本文的主要贡献在于提出了一种简单的半监督管道,该管道仅使用原始训练集而无需收集额外数据。这一方法面临两个主要挑战:1)如何仅从训练集中获取更多的训练数据;2)如何利用新生成的数据。在本研究中,我们采用了生成对抗网络(GAN)来生成未标记样本,并提出了针对异常值的标签平滑正则化(LSRO)。该方法为未标记图像分配均匀的标签分布,从而对监督模型进行正则化并提升基线性能。我们在一个实际问题上验证了所提出的方法:行人重识别(re-ID)。该任务的目标是从其他摄像机中检索查询人物。我们采用深度卷积生成对抗网络(DCGAN)进行样本生成,并使用基线卷积神经网络(CNN)进行表征学习。实验结果表明,加入由GAN生成的数据有效提升了所学CNN嵌入的判别能力。在三个大规模数据集Market-1501、CUHK03和DukeMTMC-reID上,我们的方法分别在第一级精度上比基线CNN提高了+4.37%、+1.6%和+2.46%。此外,我们将所提出的方法应用于细粒度鸟类识别任务,并在强大的基线上实现了+0.6%的性能提升。代码已发布在https://github.com/layumi/Person-reID_GAN。

代码仓库

AI-NERC-NUPT/DDB
pytorch
GitHub 中提及
AI-NERC-NUPT/PLR-OSNet
pytorch
GitHub 中提及
AI-NERC-NUPT/PFH-OSNet
pytorch
GitHub 中提及
hbchen121/dgreid
pytorch
GitHub 中提及
lyy973/OSFA
pytorch
GitHub 中提及
freeSubmission/SDB
pytorch
GitHub 中提及
layumi/Person-reID_GAN
官方
tf
GitHub 中提及

基准测试

基准方法指标
fine-grained-image-classification-on-cub-200-1Basel.+LSRO
Accuracy: 84.4
person-re-identification-on-cuhk03VI+LSRO 3
MAP: 87.4
Rank-1: 84.6
person-re-identification-on-dukemtmc-reidGAN
Rank-1: 67.68
mAP: 47.13
person-re-identification-on-market-1501GAN
Rank-1: 83.97
mAP: 66.07

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
由GAN生成的未标记样本在体外改善了行人重识别基线模型 | 论文 | HyperAI超神经