如何在Hi3559A上运行自己的YOLOv3模型(三)

caffe模型转化为海思可以识别的.wk文件

前言:

在上一篇中,我们应该已经获得了由自己的yolo模型转化而来的caffe模型文件。本篇将介绍如何将caffe model转化为.wk文件。

 

step1.

      首先打开Ruyi Stdio,工作目录可以随便选择,不要由中文就好。

step2. 

      加载官方提供的sample代码,选择file - import  如下

                                  如何在Hi3559A上运行自己的YOLOv3模型(三)_第1张图片

选择next,选择sample所在的文件夹,如下

如何在Hi3559A上运行自己的YOLOv3模型(三)_第2张图片

Select All - Finish,便完成了对于例程的加载 

step3.

            选择yolov3_func.cfg,右键,open with - Mapper Configuration Editor

          如何在Hi3559A上运行自己的YOLOv3模型(三)_第3张图片

在右侧窗口中填写你的caffe模型所在的目录。

如何在Hi3559A上运行自己的YOLOv3模型(三)_第4张图片

其余各参数的作用与意义请自行查阅《HiSVP 开发指南》,其余不修改会影响精度,不会影响实验成功

注意:marked_prototxt是选择自己的caffe model后自动生成的并更新的,无法自己填写。理论上是每次选择caffe文件后都会自动生成一个并更新过去,有时软件问题会生成文件但无法自动更新,此时需要将.cfg文件 open with -  text, 在文件的第一行自己填写在mark_prototxt下最新生成的结构文件,如图

                                          如何在Hi3559A上运行自己的YOLOv3模型(三)_第5张图片

step4. 

在所有都填写完成并保存后,点击转化按钮,即可完成模型的转化

                                          

等待转化完成,便得到了一个可以用来仿真的.wk文件。

转化过程中输出如下:

如何在Hi3559A上运行自己的YOLOv3模型(三)_第6张图片

 

下一篇:完成网络的仿真与仿真代码简要分析

       https://blog.csdn.net/qq_34533248/article/details/102497297

 

 

 

 

 

 

 

 

你可能感兴趣的:(如何在Hi3559A上运行自己的YOLOv3模型(三))