This repo is the official implementation of "Learning Semantic-Aligned Feature Representation for Text-based Person Search" in PyTorch.

Dependencies

• Python 3.7
• Pytorch 1.8.1 & torchvision 0.9.1
• numpy
• scipy 1.2.1
• pytorch_transformers

Data Preparation

1. Please download CUHK-PEDES dataset .
2. Put reid_raw.json under project_directory/data/
3. run data.sh
4. Copy files test_reid.json, train_reid.json and val_reid.json under CUHK-PEDES/data/ to project_directory/cuhkpedes/processed_data/
5. Download ViT-B_16, bert-base-uncased model and vocabulary to project_directory/pretrained_models/

Acknowledgement

Our code is extended from the following repositories. We thank the authors for releasing the codes.

• NAFS
• CMPM+CMPC