Mask Transfiner实例分割算法运行环境的搭建
Mask Transfiner是一个能实现高精度实例分割的算法,下面介绍运行代码的步骤。1、虚拟机中创建Ubuntu系统 网上关于安装Ubuntu的教程挺多的,这里只是简单示意。注意系统磁盘大小设置为40G,预防后面运行存储不够。2、Ubuntu系统安装Anaconda环境(1)下载在Ubuntu中打开Firefox,输入以下网址下载Anaconda Linux安装包https://www.
目录
一、算法原理
- 简介
Mask Transfiner是高质量和高效的实例分割算法。Mask Transfiner不是在规则的稠密张量上操作,而是将图像区域分解并表示为四叉树。该方法基于Transformer,只处理检测到的易出错的树节点,并同时进行自校正。虽然这些稀疏像素只占总数的很小比例,但它们对最终的掩模质量至关重要。这使得Mask Transfiner能够以较低的计算代价预测高度精确的实例掩码。大量实验表明,Mask Transfiner在三个流行的基准测试集上都优于当前的实例分割方法。
论文网址: https://arxiv.org/abs/2111.13673.
代码网址:https://github.com/SysCV/transfiner
- 算法框图
二、算法代码运行(跑预训练模型)
下面介绍运行代码的步骤。经过尝试,只在Ubuntu中运行成功过,在Windows中没有跑成功。
1、虚拟机中创建Ubuntu系统
网上关于安装Ubuntu的教程挺多的,这里只是简单示意。注意系统磁盘大小设置为40G,预防后面运行存储不够。
2、Ubuntu系统安装Anaconda环境
(1)下载
在Ubuntu中打开Firefox,输入以下网址下载Anaconda Linux安装包。
(2)安装
- 打开Anaconda文件所在目录,然后右键打开控制台
- 运行sh文件
bash XXX.sh
tips:XXX.sh输入前几个字母,按tab键会自动补全
期间要按Enter,还要输入yes确认。
3、Ubuntu系统安装Pycharm
(1)下载
在Ubuntu中打开Firefox,输入以下网址下载Pycharm Linux安装包。
(2)安装
打开pycharm安装包的文件夹,然后将安装包剪切,在Home目录下新建Program_Files文件,将pycharm安装包放到这里。
运行以下代码解压。
tar xzvf pycharm-community-2022.1.tar.gz
然后删除压缩包,进入解压到的文件夹的bin目录下,右键打开控制台,运行以下命令启动pycharm。
./pycharm.sh
4、pycharm创建项目
创建新项目,名为MaskTransfiner,虚拟环境选择coda,python版本选择3.7。
删除新项目的main.py文件,然后复制我整理的代码(稍后会分享链接)到项目中。
链接:https://pan.baidu.com/s/1s-xbhbsp5CUxWOp4_ESmQw
提取码:5bgp
5、安装相关配置
打开README.md,找到运行要求。
在pycharm下的Terminal运行指令。
逐个执行完以下命令。
耐心等待很长时间…
为main_test.py创建执行环境
尝试运行main_test,会发现跳出缺少包的错误
在pycharm的Terminal运行以下命令解决
pip install 包名
同样的方法解决以下错误
但是当安装pycocotools时,如果出现以下问题
我们先运行
sudo apt update
sudo apt install build-essential
后再运行
pip install pycocotools
再尝试运行main_test.py会出现以下错误
ImportError: cannot import name '_C' from 'detectron2'
我们运行以下命令解决
python setup.py build develop
6、运行效果
我们再次运行main_test.py,发现可以运行成功了!
更多推荐
所有评论(0)