Skip to content

Support most of operator which convert mxnet to caffe.

Notifications You must be signed in to change notification settings

EEWenbinWu/advanced-mxnet2caffe

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Advanced-Mxnet2Caffe

Operator Support Lists

  • Convolution
  • ChannelwiseConvolution
  • BatchNorm
  • Activation
  • ElementWiseSum
  • _Plus
  • Concat
  • Crop
  • Pooling
  • Flatten
  • FullyConnected
  • SoftmaxOutput&SoftmaxFocalOutput
  • SoftmaxActivation
  • LeakyReLU
  • elemwise_add
  • UpSampling
  • Deconvolution
  • Clip
  • Reshape

Tested models

  • Mxnet-SSH
  • MobileNet-V2
  • Resnet-50
  • RetinaFace with ( Resnet-50 ,Mobilenet 0.25 backbone)
  • All models from Insightface Model Zoo .

Note&Bugs

The convertor Is not fully automatically, The convertor not

  • if you wanna convert upsampling operator , the convertor will convert Upsampling operator to Deconvolution in Caffe , The Deconvolution channels need to be set (in prototxt_basic.py names_output).
  • If you use Flatten Layer ,You need to manually to connect them becasuse the converted compute graph will be divided into two parts.
  • If convert a detection model. You need to remove the anchor process and put it into post process.
  • Usually,If you find that conversion errors, please set the prefix name of you backbone network in mxnet2caffe.py.

About

Support most of operator which convert mxnet to caffe.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%