- Linux or macOS with python ≥ 3.6
- PyTorch ≥ 1.6
- torchvision that matches the Pytorch installation. You can install them together at pytorch.org to make sure of this.
- yacs
- Cython (optional to compile evaluation code)
- tensorboard (needed for visualization):
pip install tensorboard - gdown (for automatically downloading pre-train model)
- sklearn
- termcolor
- tabulate
- faiss
pip install faiss-cpu
conda create -n fastreid python=3.7
conda activate fastreid
conda install pytorch==1.6.0 torchvision tensorboard -c pytorch
pip install -r docs/requirements.txtPlease check the docker folder