Skip to content
/ MMN Public

Pytorch code for Towards a Unified Middle Modality Learning for Visible-Infrared Person Re-Identification

Notifications You must be signed in to change notification settings

ZYK100/MMN

Repository files navigation

Towards-a-Unified-Middle-Modality-Learning-for-Visible-Infrared-Person-Re-Identification

Paper

This repository is Pytorch code for our proposed MMN method for Cross-Modality Person Re-Identification.

Training.

Train a model by

python train.py --dataset sysu
  • --dataset: which dataset "sysu" or "regdb".

Result.

The results may have some fluctuation, and might be better by finetuning the hyper-parameters.

Datasets Rank@1 mAP
#RegDB[1] ~ 91.6% ~ 84.1%
#SYSU-MM01[2] ~ 70.6% ~ 66.9%

Citation

Please kindly cite this paper in your publications if it helps your research:

@inproceedings{zhang2021towards,
  title={Towards a Unified Middle Modality Learning for Visible-Infrared Person Re-Identification},
  author={Zhang, Yukang and Yan, Yan and Lu, Yang and Wang, Hanzi},
  booktitle={Proceedings of the 29th ACM International Conference on Multimedia},
  pages={788--796},
  year={2021}
}

Our code is based on mangye16 [3, 4].

References.

[1] D. T. Nguyen, H. G. Hong, K. W. Kim, and K. R. Park. Person recognition system based on a combination of body images from visible light and thermal cameras. Sensors, 17(3):605, 2017.

[2] A. Wu, W.-s. Zheng, H.-X. Yu, S. Gong, and J. Lai. Rgb-infrared crossmodality person re-identification. In IEEE International Conference on Computer Vision (ICCV), pages 5380–5389, 2017.

[3] M. Ye, J. Shen, G. Lin, T. Xiang, L. Shao, and S. C., Hoi. Deep learning for person re-identification: A survey and outlook. IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI), 2020.

[4] M. Ye, X. Lan, Z. Wang, and P. C. Yuen. Bi-directional Center-Constrained Top-Ranking for Visible Thermal Person Re-Identification. IEEE Transactions on Information Forensics and Security (TIFS), 2019.

If you have any question, please feel free to contact us. [email protected].

About

Pytorch code for Towards a Unified Middle Modality Learning for Visible-Infrared Person Re-Identification

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages