【MindStudio训练营第一期】MindStudio AI应用推理开发流程详解-雨天图像增强系统

环境

云主机：华为云ai1s
推理卡：Ascend 310
内存：8G
操作系统：Ubuntu 18

MindStudio下载安装

安装手册地址：https://www.hiascend.com/document/detail/zh/mindstudio/50RC3/instg/instg_000002.html

MindStudio下载地址:https://www.hiascend.com/software/mindstudio/download

下载安装脚本命令：

wget 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/MindStudio/MindStudio%205.0.RC3/MindStudio_5.0.RC3_linux.tar.gz'
tar -xvzf MindStudio_5.0.RC3_linux.tar.gz

运行MindStudio

cd MindSutdio/bin
./MindStudio.sh

MindStudio 启动界面

模型地址

DeRain

介绍

雨天图像增强技术是指将雨天成像的退化图像复原为高质量的清晰图像，消除雨滴等退化因素对图像内容的影响，其对提升，包括自动驾驶和视频监控等，户外视觉系统在恶劣天候下工作的稳定性具有重大意义。本项目基于深度学习，提出了一种去除雨天图像退化并且更好的保存图像结构的方法，并将提出的算法在华为Altas 200 DK平台上实现算法的移植。

模型地址

https://gitee.com/ascend/samples/tree/master/cplusplus/level2_simple_inference/6_other/DeRain

原始图片

推理过程

第一步：下载sample仓库

git clone https://gitee.com/ascend/samples

第二步：进入应用文件夹

cd samples/cplusplus/level2_simple_inference/6_other/DeRain/

第三步：下载图像数据

cd data
wget 'https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/DeRain/test_image/001_in.png'

第四步：下载原始模型文件

cd model
wget https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/DeRain/frozen_graph_noDWT_V2.pb
wget https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/DeRain/insert_op.cfg

第五步：进入MindStudio打开项目

点击【Open】打开下载的DeRain项目文件夹

打开后的界面

可能会有个对话框需要确认一下CANN版本，点击OK后的界面

第五步：转换模型
菜单栏【Asacnd】-【Model Converter】

Model File选择前面下载的模型文件
Output Path 选择项目中的Model文件夹路径
degradated_image改为: 1.256,256,1

然后下一步

Additional Arguments打开，然后复制–insert_op_conf=./insert_op.cfg 到框中，然后直接finish

看到ATC Start working now就说明开始进行模型转换操作了,转化完成后查看一下model 目录下的文件看一下有没有om文件

随后将om文件改名为：DeRain.om

第六步：编译应用
菜单栏【Build】-【Edit Configurations】，在出来的对话框中点+号进行设置

设置完成后点OK

菜单栏【Build】-【Build CMake Project】开始编译

第六步：运行应用

cd scripts
chmod a+x sample_run.sh
./sample_run.sh

运行结果：
进入out文件夹下output文件夹下查看输出图片