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

Mask Transfiner实例分割算法运行环境的搭建_第1张图片

目录

  • 一、算法原理
  • 二、算法代码运行(跑预训练模型)
    • 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

  • 算法框图

Mask Transfiner实例分割算法运行环境的搭建_第2张图片

Mask Transfiner实例分割算法运行环境的搭建_第3张图片 Mask Transfiner实例分割算法运行环境的搭建_第4张图片
Mask Transfiner实例分割算法运行环境的搭建_第5张图片 Mask Transfiner实例分割算法运行环境的搭建_第6张图片
Mask Transfiner实例分割算法运行环境的搭建_第7张图片 Mask Transfiner实例分割算法运行环境的搭建_第8张图片

二、算法代码运行(跑预训练模型)

    下面介绍运行代码的步骤。经过尝试,只在Ubuntu中运行成功过,在Windows中没有跑成功。

1、虚拟机中创建Ubuntu系统

Mask Transfiner实例分割算法运行环境的搭建_第9张图片
    网上关于安装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,找到运行要求。
Mask Transfiner实例分割算法运行环境的搭建_第10张图片
    在pycharm下的Terminal运行指令。
Mask Transfiner实例分割算法运行环境的搭建_第11张图片
    逐个执行完以下命令。
Mask Transfiner实例分割算法运行环境的搭建_第12张图片
耐心等待很长时间…

    为main_test.py创建执行环境
Mask Transfiner实例分割算法运行环境的搭建_第13张图片

    尝试运行main_test,会发现跳出缺少包的错误
Mask Transfiner实例分割算法运行环境的搭建_第14张图片
在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,发现可以运行成功了!
Mask Transfiner实例分割算法运行环境的搭建_第15张图片

你可能感兴趣的:(图像识别,python,图像识别,Transformer,Mask,Transfiner)