htshinichi / caffe-onnx

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

self.debug does not exist #14

Open LiyouZhou opened 4 years ago

LiyouZhou commented 4 years ago

https://github.com/htshinichi/caffe-onnx/blob/master/src/caffe2onnx.py#L481

run into an issue here that self.debug does not exist in the class.

ucalyptus commented 4 years ago

@LiyouZhou @htshinichi same issue here.

ucalyptus commented 4 years ago

@LiyouZhou I also tried removing the if else statement, But that gave me this 添加参数upsample_16_W输入信息和tensor数据 添加参数upsample_16_b输入信息和tensor数据 upsample_16 节点构建完成 Failed at layer concat, layer's bottom not detected ... 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 "/content/caffe-onnx/src/caffe2onnx.py", line 28, in init self.getNodeList(LayerList) File "/content/caffe-onnx/src/caffe2onnx.py", line 395, in getNodeList Concat_node = op.createConcat(Layers[i], nodename, inname, outname, input_shape) File "/content/caffe-onnx/src/OPs/Concat.py", line 21, in createConcat output_shape = getConcatOutShape(input_shape) File "/content/caffe-onnx/src/OPs/Concat.py", line 12, in getConcatOutShape n,c,w,h = input_shape[0][0],0,input_shape[0][2],input_shape[0][3] IndexError: list index out of range

LiyouZhou commented 4 years ago

workaround, delete these two lines. https://github.com/htshinichi/caffe-onnx/blob/master/src/caffe2onnx.py#L481-L482

ucalyptus commented 4 years ago

@LiyouZhou did, some other issue crept up.