mirror of
https://github.com/KaiyangZhou/deep-person-reid.git
synced 2025-06-03 14:53:23 +08:00
deep-person-reid
This repo contains pytorch implementations of deep person re-identification approaches.
We will actively maintain this repo.
Pretrained models
Prepare data
Train
Test
Results
xent: cross entropy + label smoothing regularizer [5] htri: triplet loss with hard positive/negative mining [4]
Market1501
Model | Size (M) | Loss | Rank-1 / -5 / -10 (%) | mAP (%) | Reported Rank | Reported mAP |
---|---|---|---|---|---|---|
ResNet50 [1] | 25.05 | xent | 85.8 / 94.4 / 96.3 | 70.1 | ||
ResNet50M [2] | 30.01 | xent | 88.8 / 95.3 / 97.0 | 74.4 | 89.9 / - / - | 75.6 |
DenseNet121 [3] | 7.72 | xent |
References
[1] He et al. Deep Residual Learning for Image Recognition. CVPR 2016. [2] Yu et al. The Devil is in the Middle: Exploiting Mid-level Representations for Cross-Domain Instance Matching. arXiv:1711.08106. [3] Huang et al. Densely Connected Convolutional Networks. CVPR 2017. [4] Hermans et al. In Defense of the Triplet Loss for Person Re-Identification. arXiv:1703.07737. [5] Szegedy et al. Rethinking the Inception Architecture for Computer Vision. CVPR 2016.
Description
Torchreid: Deep learning person re-identification in PyTorch.
computer-visioncross-domaindeep-learningdeep-neural-networksimage-retrievalmachine-learningmetric-learningperson-reidperson-reidentificationpytorchre-ranking
Readme
MIT
111 MiB
Languages
Python
99.2%
Cuda
0.4%
C++
0.2%
Dockerfile
0.1%