1. 背景介绍
- detectron2项目地址:https://github.com/facebookresearch/detectron2
- FasterRCNN论文:https://arxiv.org/abs/1506.01497
- COCO数据集:http://cocodataset.org/#home
- VOC数据集:http://host.robots.ox.ac.uk/pascal/VOC/
2. 环境要求
- Ubuntu 18.04
- CUDA10
- pytorch==1.3.0
- torchvision==0.4.1
- pycocotools
3. 安装步骤
3.1安装依赖
- cocoapi
git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install
- Cython
pip install cython
3.2 安装detectron2
git clone https://github.com/facebookresearch/detectron2.git
cd detectron2 && python -m pip install -e .
4. 测试安装
完成安装后可输入一下命令测试安装是否成功。
python3 demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \
--input cat.jpg \
--output result_cat.jpg \
--opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl
输入图片为:
结果图片为:
测试成功,说明安装成功,可以使用该框架进行进一步训练、测试等。