deep-person-reid/README.md

32 lines
1.6 KiB
Markdown
Raw Normal View History

2018-03-12 18:29:35 +08:00
# deep-person-reid
2018-03-12 18:56:27 +08:00
This repo contains [pytorch](http://pytorch.org/) implementations of deep person re-identification approaches.
2018-03-12 18:29:35 +08:00
2018-03-12 18:56:27 +08:00
We will actively maintain this repo.
2018-03-12 18:29:35 +08:00
2018-03-12 18:56:27 +08:00
## Pretrained models
## Prepare data
## Train
## Test
2018-03-12 18:29:35 +08:00
## Results
2018-03-12 19:04:39 +08:00
### Setup
* Image size: 256-by- 128 <br />
2018-03-12 19:06:31 +08:00
* Optimizer: Adam [6] <br />
2018-03-12 19:04:39 +08:00
* Loss functions: <br />
2018-03-12 19:00:14 +08:00
xent: cross entropy + label smoothing regularizer [5] <br />
htri: triplet loss with hard positive/negative mining [4] <br />
2018-03-12 19:04:39 +08:00
2018-03-12 18:29:35 +08:00
### Market1501
2018-03-12 18:56:27 +08:00
| 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 | | | | |
2018-03-12 18:29:35 +08:00
2018-03-12 18:56:27 +08:00
## References
2018-03-12 18:59:35 +08:00
[1] [He et al. Deep Residual Learning for Image Recognition. CVPR 2016.](https://arxiv.org/abs/1512.03385)<br />
[2] [Yu et al. The Devil is in the Middle: Exploiting Mid-level Representations for Cross-Domain Instance Matching. arXiv:1711.08106.](https://arxiv.org/abs/1711.08106) <br />
[3] [Huang et al. Densely Connected Convolutional Networks. CVPR 2017.](https://arxiv.org/abs/1608.06993) <br />
[4] [Hermans et al. In Defense of the Triplet Loss for Person Re-Identification. arXiv:1703.07737.](https://arxiv.org/abs/1703.07737) <br />
2018-03-12 19:06:31 +08:00
[5] [Szegedy et al. Rethinking the Inception Architecture for Computer Vision. CVPR 2016.](https://arxiv.org/abs/1512.00567) <br />
[6] [Kingma and Ba. Adam: A Method for Stochastic Optimization. ICLR 2015.](https://arxiv.org/abs/1412.6980) <br />