This repo is based on detectron2 framework.
- Linux or macOS with Python ≥ 3.6
- PyTorch ≥ 1.3
- torchvision that matches the PyTorch installation. You can install them together at pytorch.org to make sure of this.
- OpenCV, optional, needed by demo and visualization
After having the above dependencies and gcc & g++ ≥ 5, run:
# install it from a local clone:
git clone https://github.com/dragonlee258079/Saliency-Ranking.git
cd Saliency-Ranking && python -m pip install -e .
# Or if you are on macOS
# CC=clang CXX=clang++ python -m pip install -e .
To rebuild detectron2 that's built from a local clone, use rm -rf build/ **/*.so
to clean the
old build first. You often need to rebuild detectron2 after reinstalling PyTorch.