search

htshinichi / caffe-onnx

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

Upsample : Key Error #9

Closed vinaykumarngitub closed 4 years ago

vinaykumarngitub commented 4 years ago

I am trying to convert a caffemodel which consists of Upsample layer in it. But I am getting the following error : (scenario: I have un-commented the line numbers from 368-385 in caffe2onnx.py script) Traceback (most recent call last): File "convert2onnx.py", line 42, in main(args) File "convert2onnx.py", line 35, in main c2o = Caffe2Onnx(graph, params, onnx_name) File "D:\VinayKumar\Programs\Task8(TRT_SegNet)\caffe-onnx-master\src\caffe2onnx.py", line 28, in init self.__getNodeList(LayerList) File "D:\VinayKumar\Programs\Task8(TRT_SegNet)\caffe-onnx-master\src\caffe2onnx.py", line 377, in getNodeList pname = self.addInputsTVIfromMannul(Layers[i],op_pname["Upsample"],op_ptype["Upsample"],paramshape,paramdata) KeyError: 'Upsample'

Please do the needful. Thanks in advance.

ZQPei commented 4 years ago

I have fixed the problem. git pull origin master and try again?