参考资料 | 网址/文档名 |
---|---|
1 | 《HiSVP开发指南》 |
Ruyistudio集成了NNIE的mapper和simulator两大基本功能。 mapper是文件格式转换的插件,它可将caffe的模型文件*.caffemodel转换成HISDK可以识别的*.wk数据文件。
1、导入工程
Hi3559A V100R001C02SPC031\SVP_PC\HiSVP_PC_V1.1.3.0\下提供了大量的sample工程。我们通过鼠标左键单击“File”->“Import”,在弹出的导入对话框中,选择“Existing Projects into Workspace”。
导入工程所在地址。
成功导入后可在侧栏看到具体的工程目录。
其中sample_simulation为功能仿真和指令仿真工程,sample_runtime为runtime仿真工程。
2、生成wk数据文件
海思官方提供的sample中yolov3有两个cfg文件,其中由yolov3_func.cfg生成的wk文件用于PC端的仿真,yolov3_inst.cfg生成的wk文件用于开发板。
右击你想要转换的模型所对应的cfg文件,然后选择open with 再选择mapper configuration editor,或者直接双击对应的cfg文件。
需要更具实际情况对下图中红框标注的位置进行配置。prototxt、caffemodel处选择海思提供的或者自己训练好的文件。本次生成wk文件采用的是yolov3,所以net_type处应选择CNN,is_simulation处对wk文件应用的环境进行选择(运行在开发板上选择Inst/type)。下面对生成的wk文件的地址和名字进行设置。image_type和RGB_order根据具体工程情况进行选择。
配置完成后点击Make WK进行文件的生成。