海思35系列 板上正确运行 自己的yolov3模型注意事项

一 前言

因为目标检测，用到了yolov3算法，这里我检测两种类型。需要对海思官方提供的sample进行修改。按照网上一些资料，测试后发现总是缺少一个类型的检测，

但是在PC端仿真是正确的。所以只有是sample修改过程中出现了问题。按照下面的方法修改，不会出现问题，这里我已经验证过了。

这里已两个类型为例，输入图像大小为416*416.

二 修改 sample_svp_nnie_software.h 

 

这里的7= 2 + 5（2表示类型个数，5 表示bbox参数值）

三 sample_nnie.c

函数SAMPLE_SVP_NNIE_Yolov3_SoftwareInit中：修改两个地方，如红框标注：

 

 其中：pstSoftWareParam->u32ClassNum = 2 + 1;

 u32ClassNum = pstSoftWareParam->u32ClassNum;

大概研究了后面的代码，应该是后面有个for循环，起始值是从1开始的。

 最后打印出来的结果是：

 

 和caffe-yolov3对比，一下结果是正确的：