Skip to content

Latest commit

 

History

History
24 lines (20 loc) · 1018 Bytes

INSTALL.md

File metadata and controls

24 lines (20 loc) · 1018 Bytes
Raw

Installation

Requirements

  • Linux or macOS with Python ≥ 3.6
  • PyTorch ≥ 1.7 and torchvision that matches the PyTorch installation. Install them together at pytorch.org to make sure of this. Note, please check PyTorch version matches that is required by Detectron2.
  • Detectron2: follow Detectron2 installation instructions.
  • OpenCV is optional but needed by demo and visualization
  • pip install -r requirements.txt

An example of installation is shown beloe:

conda create -n zegformer python==3.7
conda activate zegformer
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
python -m pip install detectron2 -f \
  https://dl.fbaipublicfiles.com/detectron2/wheels/cu111/torch1.9/index.html

git clone https://github.com/dingjiansw101/ZegFormer.git
cd ZegFormer
pip install -r requirements.txt