海思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) 19090610466402
begin net parsing....

**********************************************************
ERROR: file: parseInputDim  line: 2244
input shape size is not equal to input name size!



**********************************************************
ERROR: file: parseInputParamOldVersion  line: 2151
ERROR: RUN FAILURE in line 2151. Return = 262148



**********************************************************
ERROR: file: parseModel  line: 14827
ERROR: RUN FAILURE in line 14827. Return = 262148



**********************************************************
ERROR: file: compile  line: 609
ERROR: RUN FAILURE in line 609. Return = 262148



**********************************************************
ERROR: file: iFEPCompile  line: 114
ERROR: RUN FAILURE in line 114. Return = 262148



**********************************************************
ERROR: file: main  line: 164
iFEPCompile 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: 1
input_dim: 3
input_dim: 608
input_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) 19090610466402
begin net parsing....
end net parsing
begin prev optimizing....
end prev optimizing....
begin net quantalizing(CPU)....
end quantalizing
begin optimizing....
end optimizing
begin NNIE[0] mem allocation....
end NNIE[0] memory allocating
begin NNIE[0] instruction generating....
end NNIE[0] instruction generating
begin parameter compressing....
end parameter compressing
begin compress index generating....
end compress index generating
begin binary code generating....
end binary code generating
begin quant files writing....
end quant files writing
wuyy@wuyy-VirtualBox:~/dl/weights$ ls
cnn_net_tree.dot          dog_bike_car.jpg    instruction.txt   yolov3_416.caffemodel  yolov3_608_inst.wk
cnn_net_tree_gv.dot       dout_nne_para.txt   mapper_debug.log  yolov3_416.cfg         yolov3_608.prototxt
cnn_net_tree_org.dot      gfpq_param.txt      mapper_error.log  yolov3_416.prototxt
cnn_net_tree_pre.dot      ibd_para.txt        netmem_info.data  yolov3_608.caffemodel
cnn_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"怎么辦就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。