README.md

April 9, 2022 ยท View on GitHub

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.