1.caffe模型加载完成
添加模型输入信息
Traceback (most recent call last):
File "convert2onnx.py", line 34, in
main(args)
File "convert2onnx.py", line 27, in main
c2o = Caffe2Onnx(graph, params, onnx_name)
File "/home/yt/software/caffe-onnx-master/src/caffe2onnx.py", line 28, in init
self.getNodeList(LayerList)
File "/home/yt/software/caffe-onnx-master/src/caffe2onnx.py", line 225, in getNodeList
conv_pname = self.addInputsTVIfromParams(Layers[i],op_pname["Conv"],op_ptype["Conv"])
File "/home/yt/software/caffe-onnx-master/src/caffe2onnx.py", line 144, in addInputsTVIfromParams
p_t = helper.make_tensor(ParamName[i],ParamType[i],ParamShape[i],ParamData[i])
File "/home/yt/.conda/envs/inface/lib/python3.8/site-packages/onnx/helper.py", line 384, in make_tensor
raise ValueError(
ValueError: Number of values does not match tensor's size. Expected 1, but it is 14112.
1.caffe模型加载完成 添加模型输入信息 Traceback (most recent call last): File "convert2onnx.py", line 34, in
main(args)
File "convert2onnx.py", line 27, in main
c2o = Caffe2Onnx(graph, params, onnx_name)
File "/home/yt/software/caffe-onnx-master/src/caffe2onnx.py", line 28, in init
self.getNodeList(LayerList)
File "/home/yt/software/caffe-onnx-master/src/caffe2onnx.py", line 225, in getNodeList
conv_pname = self.addInputsTVIfromParams(Layers[i],op_pname["Conv"],op_ptype["Conv"])
File "/home/yt/software/caffe-onnx-master/src/caffe2onnx.py", line 144, in addInputsTVIfromParams
p_t = helper.make_tensor(ParamName[i],ParamType[i],ParamShape[i],ParamData[i])
File "/home/yt/.conda/envs/inface/lib/python3.8/site-packages/onnx/helper.py", line 384, in make_tensor
raise ValueError(
ValueError: Number of values does not match tensor's size. Expected 1, but it is 14112.