2020-05-28 20:01:20 +08:00
|
|
|
# Installation
|
|
|
|
|
|
|
|
|
|
## Requirements
|
|
|
|
|
|
|
|
|
|
- Linux or macOS with python ≥ 3.6
|
2020-08-21 11:20:00 +08:00
|
|
|
- PyTorch ≥ 1.6
|
2020-06-02 10:19:20 +08:00
|
|
|
- torchvision that matches the Pytorch installation. You can install them together at [pytorch.org](https://pytorch.org/) to make sure of this.
|
2020-05-28 20:01:20 +08:00
|
|
|
- [yacs](https://github.com/rbgirshick/yacs)
|
2020-06-02 10:19:20 +08:00
|
|
|
- Cython (optional to compile evaluation code)
|
|
|
|
|
- tensorboard (needed for visualization): `pip install tensorboard`
|
2020-08-04 15:56:36 +08:00
|
|
|
- gdown (for automatically downloading pre-train model)
|
|
|
|
|
- sklearn
|
|
|
|
|
- termcolor
|
|
|
|
|
- tabulate
|
2020-08-12 16:27:57 +08:00
|
|
|
- [faiss](https://github.com/facebookresearch/faiss) `pip install faiss-cpu`
|
