Convert pytorch to Caffe by ONNX

This tool converts pytorch model to Caffe model by ONNX
only use for inference

Dependencies

• caffe (with python support)
• pytorch 0.4 (optional if you only want to convert onnx)
• onnx

we recomand using protobuf 2.6.1 and install onnx from source

git clone --recursive https://github.com/onnx/onnx.git
cd onnx 
python setup.py install

How to use

run test.py to make sure it has been installed correctly
To convert onnx model to caffe:

python convertCaffe.py ./model/MobileNetV2.onnx ./model/MobileNetV2.prototxt ./model/MobileNetV2.caffemodel

Current support operation

• Conv
• ConvTranspose
• BatchNormalization
• MaxPool
• AveragePool
• Relu
• Sigmoid
• Dropout
• Gemm (InnerProduct only)
• Add
• Mul
• Reshape
• Upsample
• Concat
• Flatten

TODO List

• support all onnx operations (which is impossible)
• merge batchnormization to convolution
• merge scale to convolution