The installation follows Detectron2. Here we provide a quickstart guide, and refer to the solutions from Detectron2 should any issue arise.
- Linux or macOS with Python ≥ 3.6
- PyTorch ≥ 1.6 and torchvision that matches the PyTorch installation
- OpenCV is optional for training and inference, yet is needed by our demo and visualization
# environment
conda create -n regionclip python=3.9
source activate regionclip
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
# RegionCLIP
git clone [email protected]:microsoft/RegionCLIP.git
python -m pip install -e RegionCLIP
# other dependencies
pip install opencv-python timm diffdist h5py sklearn ftfy
pip install git+https://github.com/lvis-dataset/lvis-api.git
To rebuild, use rm -rf build/ **/*.so to clean the old build first. You often need to rebuild detectron2 after reinstalling PyTorch.