Skip to content

aRay643/TokenMatcher

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

68 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TokenMatcher: Diverse Tokens Matching for Unsupervised Visible-Infrared Person Re-Identification

This is a code implementation and supplement to TokenMatcher: Diverse Tokens Matching for Unsupervised Visible-Infrared Person Re-Identification, which was accepted by AAAI 2025.

Contributions

  1. We introduce a TokenMatcher, a novel framework designed to extract reliable cross-modality fine-grained person features, facilitating accurate cross-modality correspondences.
  2. We present the Diverse Tokens Neighbor Learning (DTNL) module, which identifies reliable neighbors. This capability allows the model to effectively capture modality-invariant and discriminative features.
  3. We propose the Homogeneous Fusion (HF) module, which aims to minimize the differences between various camera views, thereby drawing clusters with the same identity closer together.
  4. Experiments on SYSU-MM01 and RegDB datasets demonstrate the superiority of our method compared with existing US-VI-ReID methods.

Prepare Datasets

Put SYSU-MM01 and RegDB dataset (run prepare_sysu.py and prepare_regdb.py to convert to market1501 format) into data/sysu and data/regdb. (Following previous work ADCA)

Prepare Pre-trained model

Following SDCL, we adopt the self-supervised pre-trained models (ViT-B/16+ICS) from Self-Supervised Pre-Training for Transformer-Based Person Re-Identification

Training

  1. sh run_train_sysu.sh
  2. sh run_train_regdb.sh

Models

Datestes Rank-1 mAP Download
SYSU-MM01 (Stage 1) model
SYSU-MM01 (All Search) 65.07% 62.79% model
RegDB (Visible to Infrared) 92.96% 86.32% model

Acknowledgment

Our implementation is mainly based on the following codebases. We gratefully thank the authors for their wonderful works.

SDCL. ADCA. TransReID. DC-former. CALR

About

Coder for TokenMatcher: Diverse Tokens Matching for Unsupervised Visible-Infrared Person Re-Identification (AAAI2025)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 99.3%
  • Shell 0.7%