海思3559AV100上运行自己训练的yolov3

1、简介

现在海思芯片上只支持将caffemodel转成wk文件,然而网上也没有caffe-yolov3,只能使用darknet-yolov3进行训练,然后将模型文件转换成caffemodel文件,再使用ruyistdio转成指令集的wk文件,就可以在板子上运行了。期间遇到的坑做一下记录。

遇到了一个是段错误(Segmentation fault)和精度损失问题,下面提供了解决方法

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

2、遇到的问题(价值所在)

2.1 出现Segmentation fault

将自己训练的darknet模型转成wk文件,在海思芯片上运行,只修改了SAMPLE_SVP_NNIE_Yolov3_SoftwareInit里面的 pstSoftWareParam->u32ClassNum,会出现下面的错误

你可能感兴趣的:(海思芯片)