Skip to content

Latest commit

 

History

History
33 lines (26 loc) · 1.01 KB

INSTALL.md

File metadata and controls

33 lines (26 loc) · 1.01 KB
Raw

Installation

Requirements

  • Linux or macOS with Python ≥ 3.6
  • PyTorch ≥ 1.8. 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.

Example conda environment setup

conda create --name detic python=3.8 -y
conda activate detic
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia

# under your working directory
git clone [email protected]:facebookresearch/detectron2.git
cd detectron2
pip install -e .

cd ..
git clone https://github.com/HDETR/H-Detic-LVIS.git 
cd H-Detic-LVIS
pip install -r requirements.txt

H-Deformable-Detr is already included in the repo under Deformable-DETR. To train models with Deformable-DETR (optional), we need to compile it

cd third_party/Deformable-DETR/models/ops
./make.sh