HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Unsupervised Pre-training for Person Re-identification

Dengpan Fu Dongdong Chen Jianmin Bao Hao Yang Lu Yuan Lei Zhang Houqiang Li Dong Chen

Unsupervised Pre-training for Person Re-identification

Abstract

In this paper, we present a large scale unlabeled person re-identification (Re-ID) dataset "LUPerson" and make the first attempt of performing unsupervised pre-training for improving the generalization ability of the learned person Re-ID feature representation. This is to address the problem that all existing person Re-ID datasets are all of limited scale due to the costly effort required for data annotation. Previous research tries to leverage models pre-trained on ImageNet to mitigate the shortage of person Re-ID data but suffers from the large domain gap between ImageNet and person Re-ID data. LUPerson is an unlabeled dataset of 4M images of over 200K identities, which is 30X larger than the largest existing Re-ID dataset. It also covers a much diverse range of capturing environments (eg, camera settings, scenes, etc.). Based on this dataset, we systematically study the key factors for learning Re-ID features from two perspectives: data augmentation and contrastive loss. Unsupervised pre-training performed on this large-scale dataset effectively leads to a generic Re-ID feature that can benefit all existing person Re-ID methods. Using our pre-trained model in some basic frameworks, our methods achieve state-of-the-art results without bells and whistles on four widely used Re-ID datasets: CUHK03, Market1501, DukeMTMC, and MSMT17. Our results also show that the performance improvement is more significant on small-scale target datasets or under few-shot setting.

Code Repositories

DengpanFu/LUPerson
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
person-re-identification-on-cuhk03Unsupervised Pre-training (ResNet50+BDB)
MAP: 79.6
Rank-1: 81.9
person-re-identification-on-dukemtmc-reidUnsupervised Pre-training (ResNet101+RK)
Rank-1: 93.99
mAP: 92.77
person-re-identification-on-dukemtmc-reidUnsupervised Pre-training (ResNet101+MGN)
Rank-1: 91.9
mAP: 84.1
person-re-identification-on-market-1501Unsupervised Pre-training (ResNet101+MGN)
Rank-1: 97
mAP: 92
person-re-identification-on-market-1501Unsupervised Pre-training (ResNet101+RK)
mAP: 96.21
person-re-identification-on-market-1501-cLUPerson
Rank-1: 32.22
mAP: 10.37
mINP: 0.29
person-re-identification-on-msmt17Unsupervised Pre-training (ResNet101+MGN)
Rank-1: 86.6
mAP: 68.8

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Unsupervised Pre-training for Person Re-identification | Papers | HyperAI