推荐一个好用的caffemodel转onnx的工具

我们开发了一套caffemodel转onnx模型的工具,方便算法在移动平台的部署。

我们只能提供编译好的可执行文件。

并提供脚本对onnx模型和原caffe模型进行正向推理,比较转换后的结果是否正常。

Input LRN Deconvolution
DunnyData 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,感谢。

你可能感兴趣的:(caffe,深度学习,人工智能,onnx)