Ubuntu下配置detectron2

1.先进入到自己的docker环境中(进到workspace_disk
在这里插入图片描述
在这里插入图片描述

2.部署OpenCV
python -m pip install opencv-python -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
在这里插入图片描述

3.检查OpenCV是否部署成功先输入python接着输入import cv2print(cv2.__version__)出现版本信息即部署成功
在这里插入图片描述

4.此时我们可以查看一下docker的状态,输入sudo systemctl status docker,出现以下提示代表docker在正常运行中(如果systemctl命令报错,试着输入此命令apt-get install systemd

在这里插入图片描述

5.安装Detectron2(注意看清自己的torch和Cuda版本,可输入docker ps查看,我的是torch1.8.0和cuda11.1)
在这里插入图片描述

访问https://github.com/facebookresearch/detectron2/blob/main/INSTALL.md#installation获取正确版本的Detectron2(每个人版本不同对应的链接不同,不要盲目的复制我的链接,自己去找对应的版本链接即可

在这里插入图片描述

复制python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu111/torch1.8/index.html开始下载
在这里插入图片描述

6.安装gcc和g++编译器输入apt-get install gcc g++
在这里插入图片描述

7.你也可以从源代码构建Detectron2 输入git clone https://github.com/facebookresearch/detectron2.git python -m pip install -e detectron2'安装Detectron2
在这里插入图片描述
8.查看一下自己docker内的cuda版本,输入nvcc --version
在这里插入图片描述
9.看一下workspace_disk下是否有Detectron2
在这里插入图片描述
10.输入pip install fvcore==0.1.1.post20200716

11.输入pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
安装两个必需的包和库

12.准备运行一下Detectron2的demo,保存一张图片命名为input1并将它放在demo文件夹下
在这里插入图片描述

在这里插入图片描述
如果无法拖入,应该是demo的文件夹权限问题,输入命令chmod 777 mod打开全部权限,问题应该就能解决

13.如果遇到这条报错代表fvcore版本过低,需要升级(如果没有可忽略此条
在这里插入图片描述
输入命令pip install --upgrade fvcore

14.这期间可能会出现一些包的版本过低的问题,升级以下就好了

15.运行python demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --input input1.jpg --output ./result.jpg --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

16.出现result.jpg结果代表Detectron2安装成功~
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值