文章目录
1软件环境配置
下载海思开发指南里尾号 020 的SDK,解压SVP_PC,解压 HiSVP_PC_V1.1.2.0
目录地址:
Hi3559A V100R001C02SPC020\SVP_PC\HiSVP_PC_V1.1.2.0\tools\nnie\windows

将 ruyi_env_setup-2.0.31 文件夹拷贝到一个较浅的目录,比如我直接拷贝到E盘目录下 E:\ruyi_env_setup-2.0.31
参考海思提供的HiSVP 开发指南.pdf的 5.1 RuyiStudio 安装 ,我使用的自动化脚本进行安装,但是到最后也有一些包没有装上。
注意事项:
- 安装MinGW-W64可以选择手动安装,这样更快一点
- 如果运行脚本的过程中卡住了,可以关闭脚本再运行一下
- 最后应该还有几个包没有装上,但是已经基本上不影响了,如果在程序运行的过程中出现python的环境缺失,直接pip 安装一下即可
2生成WK文件
解压 RuyiStudio-2.0.31.zip,直接打开 RuyiStudio.exe,设置工作区。进入软件。
选择 File->New->NNIE Project

输入工程名,选择芯片型号,选择 Empty ProJect,选择 MinGW GCC,
点击 Finish

以 yolov3_face 为例
在刚刚创建的工作区 workspace\yolov3_face 目录下,新建一个 images 文件夹,这里是 NNIE mapper 量化时需要的典型场景图片列表,建议从网络模型的测试场景随机选择 20~50 张作为参考图片进行量化,选择的图像要尽量覆盖模型的各个场景。
同样在 workspace\yolov3_face 目录下新建 model 文件夹,将之前转换好的 caffemodel 文件和 prototxt文件拷贝到该文件夹里。
点击 Ruyi界面 里的 yolov3_face.cfg,出现配置界面


配置步骤如下:
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张图片,这样就会自动生成List.txt
norm_type 选择data_scale ,数值默认
mean_file不用填写

7.点击Make WK,自动生成WK文件

8.最后会打印成功信息
