.官方安装方法地址:https://github.com/facebookresearch/detectron2/blob/master/INSTALL.md
1.环境要求:
1.Python >= 3.6
2.PyTorch 1.3
conda install pytorch=1.3 torchvision cudatoolkit=10.1
3.fvcore
pip install 'git+https://github.com/facebookresearch/fvcore'
4.pycocotools
pip install cython
pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
5.GCC>=4.9
2.创建Detectron2:
git clone git@github.com:facebookresearch/detectron2.git
cd detectron2
python setup.py build develop
3.测试:
在demo文件夹下放一张input1.jpg,开终端运行下面命令:
python demo/demo.py --config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \
--input input1.jpg \
--opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl
预测图能预测出来,恭喜你已经安装成功,后面我们就一起好好找到它的乐趣吧。
预测效果图: