Ubuntu 18.04 成功跑通 Detectron2 SOLOV2 demo（详细）_detetron2 solo2 config 下载-程序员宅基地

由于项目需要，刚刚接触 instance segmentation。经过论文的研读，就先把目标放在了SOLO V2上面，一开始安装的是mmdet版本的solo v2。但是奈何bug太多所以先转到了detectron2版本的。

SOLO V2(mmdet) https://github.com/WXinlong/SOLO

SOLO V2(detectron2) https://github.com/aim-uofa/AdelaiDet

我所连接的是学校配备的服务器，系统是Ubuntu18.04。首先conda是必须的（有时候conda是真的装不上东西不知道为何）。GCC 7.5。CUDA版本预设的是11.1，但是这个版本安装全部失败了，所以经过各种失败（不是管理员真的难受），最终成功地转到了CUDA10.1 update2的版本。一下是我进行配置Detectron2的全（bug）过程。中途遇到了一些module的版本错误，当然以下过程是对bug进行解决之后的。

conda create -n d2 python=3.7.3

conda activate d2

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'

在安装完cocoapi之后，通过进入python来进行测试是否安装成功。

首先对detectron2的环境进行编译：

conda install opencv

pip install fvcore==0.1.5.post20210415

pip install omegaconf==2.1.0.rc1

pip install importlib

git clone https://github.com/facebookresearch/detectron2.git

cd detectron2

python setup.py build develop

在编译成功之后，运行demo提示我 detectron2 0.4 需要pycocotools>=2.0.2. 所以我对此进行重新安装

pip install pycocotools==2.0.2

在这之后我并没有对整个的环境进行重新编译。

首先对demo实例中所需要的数据集进行下载（其他数据集可以在model ZOO中进行查找）：

wget https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

把下载好的model放到新建的model文件夹中。另外新建立test_img文件夹作为测试图片的文件夹。result作为输出结果的文件夹。

python demo/demo.py --config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --input test_img/111.jpg --output result --opts MODEL.WEIGHTS model/model_final_f10217.pkl

在成功安装detectron2框架之后，我们在当前detectron2文件夹下进行AdelaiDet的安装：

git clone https://github.com/aim-uofa/AdelaiDet.git

cd AdelaiDet

python setup.py build develop

如果detectron2安装成功的话，后面的AdelaiDet基本不会出错，在等一段较长地安装时间过后就会提示安装成功。让我们再使用刚才进行测试的图片，来跑SOLOV2的model.

首先对SOLOv2_R50_3x.pth进行下载，并放置于AdelaiDet下新建的model文件夹中。其余的文件夹的创建跟detectron2的一样。

SOLOV2 model下载链接：

https://cloudstor.aarnet.edu.au/plus/s/chF3VKQT4RDoEqC/download


python demo/demo.py --config-file configs/SOLOv2/R50_3x.yaml --input test_img/111.jpg --output result --opts MODEL.WEIGHTS model/SOLOv2_R50_3x.pth

demo运行成功后的结果图：

之后对于数据集训练的测试过程也会再之后放上来。

本文链接：https://blog.csdn.net/weixin_43776003/article/details/117740648

原作者删帖不实内容删帖广告或垃圾文章投诉

智能推荐

软件测试流程包括哪些内容？测试方法有哪些？_测试过程管理中包含哪些过程-程序员宅基地

文章浏览阅读2.9k次，点赞8次，收藏14次。测试主要做什么？这完全都体现在测试流程中，同时测试流程是面试问题中出现频率最高的，这不仅是因为测试流程很重要，而是在面试过程中这短短的半小时到一个小时的时间，通过测试流程就可以判断出应聘者是否合适，故在测试流程中包含了测试工作的核心内容，例如需求分析，测试用例的设计，测试执行，缺陷等重要的过程。..._测试过程管理中包含哪些过程

政府数字化政务的人工智能与机器学习应用：如何提高政府工作效率-程序员宅基地

文章浏览阅读870次，点赞16次，收藏19次。1.背景介绍政府数字化政务是指政府利用数字技术、互联网、大数据、人工智能等新技术手段，对政府政务进行数字化改革，提高政府工作效率，提升政府服务质量的过程。随着人工智能(AI)和机器学习(ML)技术的快速发展，政府数字化政务中的人工智能与机器学习应用也逐渐成为政府改革的重要内容。政府数字化政务的人工智能与机器学习应用涉及多个领域，包括政策决策、政府服务、公共安全、社会治理等。在这些领域，人工...

ssm+mysql+微信小程序考研刷题平台_mysql刷题软件-程序员宅基地

文章浏览阅读219次，点赞2次，收藏4次。系统主要的用户为用户、管理员，他们的具体权限如下：用户：用户登录后可以对管理员上传的学习视频进行学习。用户可以选择题型进行练习。用户选择小程序提供的考研科目进行相关训练。用户可以进行水平测试，并且查看相关成绩用户可以进行错题集的整理管理员：管理员登录后可管理个人基本信息管理员登录后可管理个人基本信息管理员可以上传、发布考研的相关例题及其分析，并对题型进行管理管理员可以进行查看、搜索考研题目及错题情况。_mysql刷题软件

根据java代码描绘uml类图_Myeclipse8.5下JAVA代码导成UML类图-程序员宅基地

文章浏览阅读1.4k次。myelipse里有UML1和UML2两种方式，UML2功能更强大，但是两者生成过程差别不大1.建立Test工程，如下图，uml包存放uml类图package com.zz.domain;public class User {private int id;private String name;public int getId() {return id;}public void setId(int..._根据以下java代码画出类图

Flume自定义拦截器-程序员宅基地

文章浏览阅读174次。需求：一个topic包含很多个表信息，需要自动根据json字符串中的字段来写入到hive不同的表对应的路径中。发送到Kafka中的数据原本最外层原本没有pkDay和project，只有data和name。因为担心data里面会空值，所以根同事商量，让他们在最外层添加了project和pkDay字段。pkDay字段用于表的自动分区，proejct和name合起来用于自动拼接hive表的名称为 ..._flume拦截器自定义开发 kafka

java同时输入不同类型数据,Java Spring中同时访问多种不同数据库-程序员宅基地

文章浏览阅读380次。原标题：Java Spring中同时访问多种不同数据库多样的工作要求，可以使用不同的工作方法，只要能获得结果，就不会徒劳。开发企业应用时我们常常遇到要同时访问多种不同数据库的问题，有时是必须把数据归档到某种数据仓库中，有时是要把数据变更推送到第三方数据库中。使用Spring框架时，使用单一数据库是非常容易的，但如果要同时访问多个数据库的话事件就变得复杂多了。本文以在Spring框架下开发一个Sp..._根据输入的不同连接不同的数据库