如何在Hi3559A上运行自己的YOLOv3模型(三)
caffe模型转化为海思可以识别的.wk文件
前言:
在上一篇中,我们应该已经获得了由自己的yolo模型转化而来的caffe模型文件。本篇将介绍如何将caffe model转化为.wk文件。
step1.
首先打开Ruyi Stdio,工作目录可以随便选择,不要由中文就好。
step2.
加载官方提供的sample代码,选择file - import 如下
选择next,选择sample所在的文件夹,如下
Select All - Finish,便完成了对于例程的加载
step3.
选择yolov3_func.cfg,右键,open with - Mapper Configuration Editor
在右侧窗口中填写你的caffe模型所在的目录。
其余各参数的作用与意义请自行查阅《HiSVP 开发指南》,其余不修改会影响精度,不会影响实验成功
注意:marked_prototxt是选择自己的caffe model后自动生成的并更新的,无法自己填写。理论上是每次选择caffe文件后都会自动生成一个并更新过去,有时软件问题会生成文件但无法自动更新,此时需要将.cfg文件 open with - text, 在文件的第一行自己填写在mark_prototxt下最新生成的结构文件,如图
step4.
在所有都填写完成并保存后,点击转化按钮,即可完成模型的转化
等待转化完成,便得到了一个可以用来仿真的.wk文件。
转化过程中输出如下:
下一篇:完成网络的仿真与仿真代码简要分析
https://blog.csdn.net/qq_34533248/article/details/102497297