使用MindStudio进行语义分割应用开发

• 介绍

1、 MindStudio

MindStudio是一款专为AI开发设计的代码编辑器。旨在提供满足AI开发全过程所需的一站式开发环境，支持模型开发、算子开发以及应用开发三个主流程中的开发任务。本篇文章针对应用开发这一任务，分享使用经验。

2、 模型和任务介绍

我在本次开发中使用的模型为AttU_Net。该模型使用了PyTorch框架，用于图片的语义分割。模型原作者提出了一种新的结构——注意力门（attention gate,AG）。AttU_Net会自动学习区分目标的外形和尺寸。这种有attention gate的模型在训练时会学会抑制不相关的区域，注重有用的显著特征。就像人类的视觉运作的方式一样，只会把注意力放在关键的部分，attention gate模拟了这一机制，也会给模型带来类似于人类注意力机制的能力，让模型拥有自动寻找图片中重要的位置的能力，并着重分析这一部分。

在本实例中，使用的数据集为ISIC 2018 Task 1的数据，包括2594张显微镜下拍摄的医学图片以及对应的分割结果。我这里下载的是Task 1的training data和training ground truth，也就是在下面的图片中圈出来的两个。想要使用相同数据集的话可以自行到数据集名称关联的超链接中进行下载。

 

 

    本次应用开发基于MindX SDK进行，关于MindX SDK的更多详细信息可以在昇腾官方发布的MindStudio文档的“基于MindX SDK开发应用”章节中查看。

二、 环境搭建

1、 MindStudio下载和安装

MindStudio下载连接如下：

MindStudio下载-昇腾社区 (hiascend.com)

进入页面后可以看到MindStudio的基本信息和版本介绍，往下翻可以就看到Linux系统和Windows系统的安装包，以及Windows系统下的免安装压缩包。可以根据自己的需要进行下载和安装。安装过程比较简单，同意服务协议、确认安装位置之后，等待安装完成即可。

2、 CANN下载和安装

CANN (Compute Architecture for Neural Networks)是华为公司针对AI场景推出的异构计算架构，通过提供多层次的编程接口，支持用户快速构建基于昇腾平台的AI应用和业务。CANN包含的几个部分如下图所示。使用MindStudio开发的过程需要使用CANN。

 

 

CANN安装，可以参考下面的连接，按照官方文档给出的步骤进行下载和安装。

软件安装-环境准备-5.1.RC1-CANN商用版-文档首页-昇腾社区 (hiascend.com)

下面详细展示一下在MindStudio中配置CANN的过程。

首先需要配置SSH连接到远程服务器。在Settings中找到SSH Configurations，位置如下图所示，在Tools菜单下。 如果是第一次使用MindStudio，可以在初始窗口左侧边栏中点击Customize，然后在出现的界面中点击最下方的All Settings。项目界面中可以在左上角图标旁边的File菜单中找到Settings，和PyCharm、CLion等编辑器相同。

 

 

    填写好对应信息后，点击上图中圈出的Test Connection按钮，检查连接是否成功。看到下面的小窗口在屏幕中间弹出则表示连接成功。如果提示连接失败，请检查服务器状态、网络是否正常、信息填写是否准确无误等关键部分。

   

 

    接下来是CANN的设置。找到Settings -> Appearance & Behavior -> System Settings -> CANN。进入CANN管理界面。点击右上角的文件夹Change CANN按钮。如果尚未完成CANN安装，按钮上的文字会显示为Install CANN，点击它。进入CANN设置界面。

 

 

    在这里需要填写远程服务器地址。并选择CANN所在位置。

 

 

    点击服务器地址栏右边的小加号，会弹出SSH Configuration的窗口。如下图。可以进行连接测试。