htshinichi / caffe-onnx

caffe model convert to onnx model
MIT License
175 stars 43 forks source link

ValueError: Number of values does not match tensor's size. Expected 1, but it is 14112. #37

Open Sunshine066 opened 1 year ago

Sunshine066 commented 1 year ago

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.