## Clone the repository ``` git clone https://github.com/aminebdj/OpenYOLO3D.git cd OpenYOLO3D ``` ## Conda Environment ``` export TORCH_CUDA_ARCH_LIST="6.0 6.1 6.2 7.0 7.2 7.5 8.0 8.6" conda env create -f environment.yml conda activate openyolo3d cd models/Mask3D/ pip3 install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113 pip3 install torch-scatter -f https://data.pyg.org/whl/torch-1.12.1+cu113.html pip3 install 'git+https://github.com/facebookresearch/detectron2.git@710e7795d0eeadf9def0e7ef957eea13532e34cf' --no-deps cd third_party git clone --recursive "https://github.com/NVIDIA/MinkowskiEngine" cd MinkowskiEngine git checkout 02fc608bea4c0549b0a7b00ca1bf15dee4a0b228 python setup.py install --force_cuda --blas=openblas cd .. git clone https://github.com/ScanNet/ScanNet.git cd ScanNet/Segmentator git checkout 3e5726500896748521a6ceb81271b0f5b2c0e7d2 make cd ../../ cd pointnet2 python setup.py install cd ../../ pip3 install pytorch-lightning==1.7.2 pip install black==21.4b2 cloudpickle==3.0.0 future hydra-core==1.0.5 pycocotools>=2.0.2 pydot iopath==0.1.7 loguru albumentations pip install . cd ../YOLO-World pip install -e . cd ../../ pip install mmyolo==0.6.0 mmdet==3.0.0 plyfile pip install openmim pip3 install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113 mim install mmcv==2.0.0 pip install open3d pip install pillow==9.1.0 ``` ## Prepare the class agnostic masks ``` sh scripts/get_class_agn_masks.sh ``` ## Prepare the checkpoints ``` sh scripts/get_checkpoints.sh ```