Skip to content

Latest commit

 

History

History
49 lines (35 loc) · 1.46 KB

INSTALL.md

File metadata and controls

49 lines (35 loc) · 1.46 KB
Raw

Installation steps for SoundQ

Repo setup

We recommend using conda, this eases some dependencies with cuda for running all the available submodules in this repo.

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

Start repo submodules

git submodule sync --recursive
git submodule update --init --recursive

Setup audiovisual_seld

Setup detection/

Note: adapted from Detic docs

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.

Install submodules required packages

cd detectron2
pip install -e .

cd ..
cd Detic
pip install -r requirements.txt

Our project uses two submodules, CenterNet2 and Deformable-DETR. If you forget to add --recurse-submodules, do git submodule init and then git submodule update. To train models with Deformable-DETR (optional), we need to compile it

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