Inference wrapper for Detectron2
- detectron2:
./install_detectron2_dep.sh && python3 -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
# (add --user if you don't have permission)
# Or, to install it from a local clone:
git clone https://github.com/facebookresearch/detectron2.git
cd detectron2 && python3 -m pip install -e .
# Or if you are on macOS
# CC=clang CXX=clang++ python -m pip install -e .
- cv2
- In the main project folder, install
det2
as a python package usingpip
or as an editable package if you like (-e
flag afterpip
)
cd det2 && pip3 install .
- Or, download
.whl
file in this repo's releases
- Get docker:
docker pull levan92/det2
OR Build your own docker fromDockerfile
, you will need the tensorrt installation deb file - Use docker:
./run_docker.sh
look into the script to change paths accordingly
See files in example_scripts/.
cd example_scripts/weights
./get_weights.sh