Spatio-temporal Information Mining and Fusion Feature-Guided Modal Alignment for Video-based Visible-Infrared Person Re-identification
We use /torch >=1.12.0 / 24G RTX3090 for training and evaluation.
pip install -r SIMFGL/requirements.txt
- Download BUPTCampus from baidu disk. The file structure should be:
path_to_dataset
|—— DATA
|—— data_paths.json
|—— gallery.txt
|—— query.txt
|—— train.txt
|—— train_auxiliary.txt
- Download VCM-HITSZ from baidu disk. The file structure should be:
path_to_dataset
|—— ID
|—— info
1.Train and test on the VCM-HITSZ dataset by first navigating to the VCM-HITSZ directory and running train.py or test.py.
python train.py/test.py
1.Train and test on the BUPTCampus dataset by first navigating to the BUPTCampus directory and running train.py or test.py.
python train.py/test.py
For direct testing, please download our prepared checkpoints and extracted features from baidu disk.
A large part of codes are borrowed from MITML. Thanks for their excellent work!
If you have any questions, please feel free to contact me. (zuozhigang2024@163.com).