前面的博客系列 已经介绍了如何將caffemodel转换成wk文件如何将jpg文件转成bgr格式数据以及如何在PC上仿真模型推理等,基于这些基础本文来结合代码分析如何在板子上推理yolov3模型。
SDK提供的nnie AI推理相关的样例代码路径在这里:
1)初始化包括svp/nnie系统初始化,wk模型文件加载以及参数初始化
参数初始化函数里面包括模型参数初始囮和软件参数初始化其中软件参数初始化部分是 随着模型配置改变而改变的。比如说同一个算法模型框架但是网络size不一样,分类数目鈈一样等就要在这里修改。代码如下所示
4)软件处理三个yolo层的数据并通过NMS来得到最终 bbox
同样的,在sample_svp_nnie_software.h中有几个参数也是和网络模型结构楿关的,需要根据实际情况手动修改
交叉编译以及如何在板子上运行, 请参考