模型转换工具合集与 pytorch2caffe

模型转换工具

github地址: https://github.com/ysh329/deep-learning-model-convertor
模型转换的部分截图如下：

海思自己的在线转换工具应该是使用了微软的MMdnn（https://github.com/Microsoft/MMdnn）

pytorch2caffe

尝试一（失败）

网上很多人都用的是https://github.com/longcw/pytorch2caffe
可惜pytorch版本要求低于0.4，实在是太老了，模型都加载不起来。
虽然最后失败了，但是我也把大致步骤写出来，给大家一个参考。

1.使用源码安装pytorch0.2 （实在是找不到0.2–CPU版本的安装包）

2.import torch 出现如下问题

ImportError: /usr/local/share/anaconda2/envs/onnx_env/lib/python2.7/site-packages/torch/_C.so: undefined symbol: _ZN3MPI8Datatype4FreeEv

在https://github.com/pytorch/pytorch/issues/2680里搜索，有人说有一个临时的解决办法是LD_PRELOAD=/usr/lib/libmpi_cxx.so python

这时候就需要源码编译 openmpi，编译前记得加上选项--enable-mpi-cxx，这样才会有 libmpi_cxx.so 文件。

最后打开python
LD_PRELOAD=/usr/local/openmpi/lib/libmpi_cxx.so python

加载pytorch
import torch

成功
3.最后加载模型的时候，一部分没办法加载成功。放弃！

尝试二（失败）（更新）

思路：pytorch->ONNX->Caffe
pytorch可以方便的转到ONNX（成功）
ONNX模型简化https://github.com/daquexian/onnx-simplifier（成功）
ONNX2Caffehttps: