detectron2安装教程
- Detectron是Facebook人工智能研究的软件系统,它实现了最先进的目标检测算法,包括Mask R-CNN。它是用Python编写的,由Caffe2深度学习框架提供支持。
# 1.配置虚拟环境
conda create -n detectron_env python=3.8.16
# 2.激活虚拟环境
conda activate detectron_env
# 3.安装pycocotools
pip install cython
pip install numpy
pip install "git+https://gitee.com/wsyin/cocoapi.git#subdirectory=PythonAPI"
# 4.安装cuda,pytroch,注意这里需要从官网查找,否则报错,最好按照我的版本来
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.3 -c pytorch -c conda-forge
# 5.克隆代码到远程服务器/本地
git clone https://github.com/facebookresearch/detectron2.git
# 6.进入项目
cd detectron2/
# 7.运行本地安装
pip install -e .
# 8.安装opencv,官网说可以不安装,但是这个不安装也会报错
pip install opencv-python
# 在项目目录下创建一个img文件夹,复制一张图片1.jpg到 该文件夹
8.创建img文件夹,引入1.jpg文件
# 9. 运行detectron2,跑目标检测预训练模型
python3 demo.py --config-file ../configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml --input ../img/1.jpg --output ../img/1_1.jpg --opts MODEL.WEIGHTS detectron2://COCO-Detection/faster_rcnn_R_50_FPN_3x/137849458/model_final_280758.pkl
# 10.运行detectron2,跑实例分割预训练模型
python demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --input ../img/2.jpg --output ../img/2.3.jpg --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl
按照此步骤按照已经成功!
原图
检测图片