我们开发了一套caffemodel转onnx模型的工具,方便算法在移动平台的部署。
我们只能提供编译好的可执行文件。
并提供脚本对onnx模型和原caffe模型进行正向推理,比较转换后的结果是否正常。
Input | LRN | Deconvolution |
Input | InnerProduct | Interp |
VideoData | Reshape | Split |
Convolution | Transpose | Slice |
Convolution3D | Gemm | ShuffleChannel |
ConvolutionDepthwise | DropOut | Axpy |
DepthwiseConvolution | Concat | Crop |
BatchNorm | Swish | Power |
Normalize | Sigmoid | Im2Col |
BN | Mish | Transpose |
Scale | BroadcastMul | Permute |
ReLU | Eltwise | Lstm |
PReLU | Flatten | LSTM |
Pooling | MaxUnpool | Reverse |
Pooling3D | Upsample | Reorg |
Softmax | Uppooling | SpatialProduct |
AbsVal | Threshold | Reduction |
工程网址以及使用方法请参考网址:https://github.com/xncaffe/caffe_convert_onnx
觉得好用请点star,感谢。