安装
我们的Colab Notebook(https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5) 有具体的安装指导,说明了如何安装detectron2。Dockerfile(https://github.com/facebookresearch/detectron2/blob/master/docker/Dockerfile) 还安装了detectron2常用的一些简单命令。
要求
带有Python版本≥3.6的Linux或macOS
PyTorch版本≥1.3
torchvision的版本与PyTorch的版本要相匹配。你可以在pytorch.org上将它们安装在一起来确保版本一致。
演示和可视化所需的OpenCV(可选)
pycocotools:
pip install cython; pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
从源代码构建Detectron2
在具有上述环境以及gcc&g++的版本≥4.9之后,运行:
pip install 'git+https://github.com/facebookresearch/detectron2.git'# (如果没有权限,则添加user)
# 或者,从本地克隆安装:git clone https://github.com/facebook