/caffe2onnx

caffe model to onnx

Primary LanguagePython

caffe2onnx

Supported layers

  • Convolution
  • DepthwiseConvolution
  • BatchNorm
  • Scale
  • ReLU
  • Pooling
  • Eltwise
  • InnerProduct
  • Softmax
  • Sigmoid
  • Concat
  • Deconvolution
  • Reshape
  • Permute
  • Scale
  • Log
  • Power
  • Bnll
  • Selu
  • Sqrt
  • Cos
  • CRelu
  • Abs
  • Exp
  • TanH
  • Tile
  • PReLU
  • Sin
  • Slice
  • ReLU6
  • ELU
  • Upsample
  • UpsampleBN
  • Flatten
  • Squeeze
  • Transpose
  • Lstm
  • Gru

Tested models

  • Resnet-18
  • Resnet-50
  • Mobilenet-v1
  • Mobilenet-v2
  • Yolov5