Mask Transfiner实例分割算法运行环境的搭建

目录

• 一、算法原理
• 二、算法代码运行（跑预训练模型）
• • 1、虚拟机中创建Ubuntu系统
  • 2、Ubuntu系统安装Anaconda环境
  • • （1）下载
    • （2）安装
  • 3、Ubuntu系统安装Pycharm
  • • （1）下载
    • （2）安装
  • 4、pycharm创建项目
  • 5、安装相关配置
  • 6、运行效果

一、算法原理

• 简介

    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，发现可以运行成功了！