本文主要给大家介绍使用MindStudio进行PSENet_ResNet50_vd模型离线推理开发,完成图像文本检测,主要内容包括MindStudio安装、MindStudio工程创建、模型介绍与获取、模型转换、模型推理5个部分。
一、MindStudio介绍
MindStudio是一套基于华为昇腾AI处理器开发的AI全栈开发平台,包括基于芯片的算子开发、以及自定义算子开发,同时还包括网络层的网络移植、优化和分析,另外在业务引擎层提供了一套可视化的AI引擎拖拽式编程服务,极大的降低了AI引擎的开发门槛。
本文使用MindStudio版本为:5.0.RC3,MindStudio功能框架如图1所示,集成开发环境包括:工程管理、编译、运行、模型转换、应用开发、精度比对、自定义算子、性能分析、仿真器、Debug、CANN-Toolkit功能。
图1
离线推理开发主要使用的是模型转换功能,将训练好的第三方网络模型可以直接通过离线模型工具导入并转换成离线模型,并可一键式自动生成模型接口,方便开发者基于模型接口进行编程,同时也提供了离线模型的可视化功能。
MindStudio的安装过程可以参考官方指导手册,链接为:
安装指南部分详细介绍了安装相关步骤,如图2所示。
图2
二、MindStudio工程创建
1.1 主界面功能介绍
首次启动MindStudio会提示是否导入配置,如图3所示,可以选择不导入配置,点击OK进入MindStudio主界面。
图3
MindStudio欢迎界面比较简洁,相关功能如下:
a.工程管理功能,如图4所示,包括新建工程,打开、导入工程。
图4
b.用户设置功能,可以设置主题、字体等,如图5所示,将界面颜色设置为IntelliJ Light。
图5
c.插件管理功能,如图6所示,可以安装、卸载一些常用的插件。
图6
d.学习功能,如图7所示,可以进入到MindStudio用户帮助界面。
图7
1.2 创建工程
使用MindStudio主界面的工程管理功能创建一个新的工程,具体步骤为:
a.点击New Project开始创建一个新的工程,如图8所示。
图8
b.New Project界面如图9所示,选择Ascend App,点击Change进行CANN配置。
图9
c.CANN配置界面如图10所示,点击Remote Connection的+号图标添加远程服务器链接,点击SSH Configurations中的+号,输入远程服务器相关信息,点击OK。
图10
d.如图11所示,点击Remote CANN location中的文件夹图标,选择远程服务CANN安装路径,点击OK确认,点击Finish结束配置。
图11
e.CANN配置结束后会自动同步远程CANN信息,如图12所示。
图12
f.同步完成后自动返回New Project界面,如图13所示,选择ACL Project(Python),点击Next继续。