Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

Garfield2005/Mxnet2Caffe

Repository files navigation

MXNet2Caffe: Convert MXNet model to Caffe model

UPDATE

  1. 2019年01月08日: Clone from https://github.com/GarrickLin/MXNet2Caffe.git
  • 增加log
  • 增加relu6(clip in mxnet): caffe中没有relu6,使用relu代替,之前的实验表明该替换对推断影响不大
  • 增加 Deconvolution:主要用于分割网络的最后几层
  • 增加 Upsampling:caffe中没有单独的上采样层,可以通过Deconvolution变相实现
  • 增加 SoftmaxOutput:该层在mxnet中为一层,转换到caffe后,使用两层:Softmax和Argmax

USAGE

使用环境

  1. 必须同时安装caffe环境及mxnet环境,可以使用本人做好的docker镜像:docker pull kd-bd02.kuandeng.com/kd-recog/recog:mxnet2caffe
  2. 修改脚本中mxnet的模型路径及caffe模型的保存路径,然后运行脚本即可:mxnet2caffe.sh

TODO

  1. 转换脚本中载入mxnet模型的方式不是太合理,但不影响使用

About

Convert Mxnet model to caffe model

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

AltStyle によって変換されたページ (->オリジナル) /