海思Hi3559AV100将caffemodel转为wk文件

使用ruyistudio将caffemodel转换为wk文件流程。

确保prototxt文件格式符合要求

在将cfg文件以及weights文件转换为caffemodel文件以及prototxt文件之后,prototxt文件的开头格式可能为下图:
海思Hi3559AV100将caffemodel转为wk文件_第1张图片
但这个不符合ruyistudio转wk文件格式,应将其改为:
海思Hi3559AV100将caffemodel转为wk文件_第2张图片
如果格式不正确,那么在ruyistudio转wk文件时,会提示prototxt格式错误,并且不能自动填充marked_prototxt选项。

NNIE Project

  1. 新建NNIE Project,如命名为yolov3.
    海思Hi3559AV100将caffemodel转为wk文件_第3张图片
  2. 在新工程yolov3文件夹下添加新的文件夹model以及images。其中model文件夹中放入待转的caffemodel以及prototxt文件,images文件夹下放20~50张图片。
  3. 双击yolov3.cfg文件,进行配置。海思给定自带的yolov3转wk的cfg文件配置如下图: ![simulation仿真模式,转为yolov3_fun.wk文件](https://img-blog.csdnimg.cn/2021052510212861.png?x-oss-simulation process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L20wXzQ5MDA5OTg2,size_16,color_FFFFFF,t_70)
    海思Hi3559AV100将caffemodel转为wk文件_第4张图片
    配置好cfg文件之后,点击make wk按钮,生成相应的wk文件。在这里插入图片描述

遇到的问题:

  1. 在ubuntu系统中转好的prototxt文件在Windows系统中打不开,原因如下:
    cfg文件中存在一些不必要的注释掉的文字,需要把注释掉的删除,保持跟官方cfg一致。
    修改之后,再次转prototxt文件,则可以在windows下打开。

  2. prototxt文件中文字格式必须是英文,空格也必须是英文空格,如果是中文空格,会提示有错误。

转换成YUV格式的wk文件

海思Hi3559AV100将caffemodel转为wk文件_第5张图片

你可能感兴趣的:(caffe,神经网络,深度学习)