RuyiStudio的安装和转换模型

文章目录

  • 1软件环境配置
  • 2生成WK文件

注:该博文只是极端理想的情况(模型移植精度损失不大,后处理代码也无需过多修改),可以作为熟悉大致流程的例子看看。移植模型绝对不会这么理想…

1软件环境配置

下载海思开发指南里尾号 020 的SDK,解压SVP_PC,解压 HiSVP_PC_V1.1.2.0

目录地址:
Hi3559A V100R001C02SPC020\SVP_PC\HiSVP_PC_V1.1.2.0\tools\nnie\windows
RuyiStudio的安装和转换模型_第1张图片
ruyi_env_setup-2.0.31 文件夹拷贝到一个较浅的目录,比如我直接拷贝到E盘目录下 E:\ruyi_env_setup-2.0.31
参考海思提供的HiSVP 开发指南.pdf的 5.1 RuyiStudio 安装 ,我使用的自动化脚本进行安装,但是到最后也有一些包没有装上。

注意事项:

  1. 安装MinGW-W64可以选择手动安装,这样更快一点
  2. 如果运行脚本的过程中卡住了,可以关闭脚本再运行一下
  3. 最后应该还有几个包没有装上,但是已经基本上不影响了,如果在程序运行的过程中出现python的环境缺失,直接pip 安装一下即可

2生成WK文件

解压 RuyiStudio-2.0.31.zip,直接打开 RuyiStudio.exe,设置工作区。进入软件。
选择 File->New->NNIE Project
RuyiStudio的安装和转换模型_第2张图片
输入工程名,选择芯片型号,选择 Empty ProJect,选择 MinGW GCC
点击 Finish
RuyiStudio的安装和转换模型_第3张图片
yolov3_face 为例

在刚刚创建的工作区 workspace\yolov3_face 目录下,新建一个 images 文件夹,这里是 NNIE mapper 量化时需要的典型场景图片列表,建议从网络模型的测试场景随机选择 20~50 张作为参考图片进行量化,选择的图像要尽量覆盖模型的各个场景。

同样在 workspace\yolov3_face 目录下新建 model 文件夹,将之前转换好的 caffemodel 文件和 prototxt文件拷贝到该文件夹里。

点击 Ruyi界面 里的 yolov3_face.cfg,出现配置界面

RuyiStudio的安装和转换模型_第4张图片
RuyiStudio的安装和转换模型_第5张图片
配置步骤如下:
1.点击 Browse 选择刚刚放在 model 下的两个模型文件

在这里插入图片描述
2.Net_type根据网络选择CNN,如果想要生成功能仿真的 WK 文件 is_simulation 选择Simulation,如果想要生成指令仿真或在板子上运行的WK文件,选择Inst/Chip
在这里插入图片描述
3.在output_wk_name一栏中输入生成 WK文件的名字
在这里插入图片描述
4.在第一步 “点击Browse 选择刚刚放在 model下的两个模型文件” 时,软件会自动在workspace\yolov3_face\mark_prototxt目录下生成一个新的 prototxt(即《HiSVP 开发指南.pdf》中166页Prototxt文件自动标记功能),这时需要将第一步中选择的 prototxt文件更换为软件新生成的 prototxt 文件。此时的marked_prototxt一栏会自动填上内容。
在这里插入图片描述
5.Mapper Seting配置如图
compile_mode 选择 Low-bandwidth(低精度模式int8)
log_level 选择Module level(打印信息,根据自己需要来选择)
align_bytes 选择 16
batch_num 选择 1
sparse_rate 选择 0
在这里插入图片描述
6.image_type 一般选择U8(例如我想在板子上直接推理YUV420SP格式的图片,此时就需要选择YUV420SP
RGB_order 选择 BGR
image_list 点击 Create,选择我们之前准备好的20张JPG图片,这样就会自动生成List.txt
norm_type 选择data_scale ,数值默认(预处理步骤,需要根据模型的数据预处理来选择)
mean_file不用填写(预处理步骤,需要根据模型来选择)
RuyiStudio的安装和转换模型_第6张图片

7.点击Make WK,自动生成WK文件
RuyiStudio的安装和转换模型_第7张图片
8.最后会打印成功信息
在这里插入图片描述

你可能感兴趣的:(#,海思NNIE与深度学习,ruyi,华为,NNIE)