liyuke65535 / ACL-DGReID

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Adaptive Cross-Domain learning for Generalizable Person Re-Identification

Requirements

  • CUDA>=10.0
  • At least four 1080-Ti GPUs
  • Setup could refer to INSTALL.md
  • Other necessary packages listed in requirements.txt
  • Training Data
    The model is trained and evaluated on Market-1501, MSMT17, cuhkSYSU, CUHK03. Download for these datasets, please refer to fast-reid.

Run

  # train
  python copy_launch.py

  # test
  python3 tools/train_net.py --config-file ./configs/bagtricks_DR50_mix.yml --eval-only MODEL.WEIGHTS /path/to/checkpoint_file MODEL.DEVICE "cuda:0"

Acknowledgments

This repo borrows partially from fast-reid, and MetaBIN.

About

License:Apache License 2.0


Languages

Language:Python 98.6%Language:Cython 1.4%Language:Shell 0.0%Language:Makefile 0.0%