- Classification: reid_classification.py
- Classification + triplet loss: reid_tripletcls.py (triplet_loss)
- Classification + triplet loss with hard negative mining: reid_tripletcls.py (triplet_hard_loss)
- Classification + margin sample mining loss: reid_tripletcls.py (msml_loss)
- Re-ranking with k-reciprocal Encoding (CVPR2017): re_ranking.py
- Using Pytorch (GPU) to accelerate Re-ranking with k-reciprocal Encoding (CVPR2017): re_ranking_gpu.py
- Pre-trained model: naivehard_more_last.h5 (market1501:81.1% rank-1 accuracy cuhk03:78.8% rank-1 accuracy)
http:https://pan.baidu.com/s/1bo3gwaV
You should add some functions according to the comments
@article{xiao2017margin,
title={Margin Sample Mining Loss: A Deep Learning Based Method for Person Re-identification},
author={Xiao, Qiqi and Luo, Hao and Zhang, Chi},
journal={arXiv preprint arXiv:1710.00478},
year={2017}
}