Skip to content

Margin Sample Mining Loss: A Deep Learning Based Method for Person Re-identification

Notifications You must be signed in to change notification settings

michuanhaohao/keras_reid

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Person reid for keras

  • 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)

Pre-trained model download

http://pan.baidu.com/s/1bo3gwaV

Note

You should add some functions according to the comments

Citation

@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}
}

About

Margin Sample Mining Loss: A Deep Learning Based Method for Person Re-identification

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages