Open huiofficial opened 4 years ago
Besides, I find that Ops doesn't have all the operations I need, but your doc says only to modify my caffe.proto. I wonder whether it matters.
The code uses the onnx helper.make_node to convert the caffemodel to onnx model. Before calling the helper.make_node, you should get the parameters of the function. It is necessary to define the operators you need in the OPs directory. Before you define the operator, you need to check for the operators that onnx support.
我也遇到了。他上一层是Normalize 目前代码里没实现。
I try to use your code to convert TensorRT7.0's SSD caffe model to ONNX, I modified the caffe.proto to add all the Ops and it works fine until I met Layer
conv4_3_norm_mbox_loc
.In
src/caffe-onnx.py
I find that the function__getLastLayerOutNameAndShape()
cannot read this layers output so the convert is stopped. I find you useimport ipdb; ipdb.set_trace()
to specifically deal with layerconv4_3_norm_mbox_loc
, but I don't know how to use it the right way and help me finish the convert.Your code is:
Thanks.