运行效果

一、算法原理

  • 简介

    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安装包。

https://www.anaconda.com/

在这里插入图片描述

(2)安装

  • 打开Anaconda文件所在目录,然后右键打开控制台
  • 运行sh文件

bash XXX.sh
tips:XXX.sh输入前几个字母,按tab键会自动补全

在这里插入图片描述

期间要按Enter,还要输入yes确认。

3、Ubuntu系统安装Pycharm

(1)下载

    在Ubuntu中打开Firefox,输入以下网址下载Pycharm Linux安装包。

https://www.jetbrains.com/pycharm/download

在这里插入图片描述

(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,发现可以运行成功了!
在这里插入图片描述

Logo

更多推荐