海思NNIE mapper转换模型出错提示："input shape size is not equal to input name size"怎么办

这篇文章给大家介绍海思NNIE mapper转换模型出错提示："input shape size is not equal to input name size"怎么办，内容非常详细，感兴趣的小伙伴们可以参考借鉴，希望对大家能有所帮助。

caffe模型在电脑上运行正常，使用海思nnie_mapper转换模型时失败，输出如下：

wuyy@wuyy-VirtualBox:~/dl/weights$ nnie_mapper_12 yolov3_608.cfg Mapper Version 1.2.2.1_B030 (NNIE_1.2) 19090610466402begin net parsing....**********************************************************ERROR: file: parseInputDim  line: 2244input shape size is not equal to input name size!**********************************************************ERROR: file: parseInputParamOldVersion  line: 2151ERROR: RUN FAILURE in line 2151. Return = 262148**********************************************************ERROR: file: parseModel  line: 14827ERROR: RUN FAILURE in line 14827. Return = 262148**********************************************************ERROR: file: compile  line: 609ERROR: RUN FAILURE in line 609. Return = 262148**********************************************************ERROR: file: iFEPCompile  line: 114ERROR: RUN FAILURE in line 114. Return = 262148**********************************************************ERROR: file: main  line: 164iFEPCompile failed(262148)begin quant files writing....end quant files writing

原因是deploy.prototxt文件输入层的格式与海思的nnie_mapper支持的格式不兼容，海思支持的格式参阅文档：HiSVP 开发指南.pdf，章节：3.2.1 deploy.prototxt 输入层格式 原来的格式如下：

input: "data"input_dim: 1input_dim: 3input_dim: 608input_dim: 608

修改如下：

input: "data"input_shape {        dim: 1        dim: 3        dim: 608        dim: 608}

修改后重新运行nnie_mapper，转换成功：

wuyy@wuyy-VirtualBox:~/dl/weights$ nnie_mapper_12 yolov3_608.cfg Mapper Version 1.2.2.1_B030 (NNIE_1.2) 19090610466402begin net parsing....end net parsingbegin prev optimizing....end prev optimizing....begin net quantalizing(CPU)....end quantalizingbegin optimizing....end optimizingbegin NNIE[0] mem allocation....end NNIE[0] memory allocatingbegin NNIE[0] instruction generating....end NNIE[0] instruction generatingbegin parameter compressing....end parameter compressingbegin compress index generating....end compress index generatingbegin binary code generating....end binary code generatingbegin quant files writing....end quant files writingwuyy@wuyy-VirtualBox:~/dl/weights$ lscnn_net_tree.dot          dog_bike_car.jpg    instruction.txt   yolov3_416.caffemodel  yolov3_608_inst.wkcnn_net_tree_gv.dot       dout_nne_para.txt   mapper_debug.log  yolov3_416.cfg         yolov3_608.prototxtcnn_net_tree_org.dot      gfpq_param.txt      mapper_error.log  yolov3_416.prototxtcnn_net_tree_pre.dot      ibd_para.txt        netmem_info.data  yolov3_608.caffemodelcnn_net_tree_pre_org.dot  image_ref_list.txt  RUNTIME.txt       yolov3_608.cfg

关于海思NNIE mapper转换模型出错提示："input shape size is not equal to input name size"怎么办就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。