pytorch训练的模型转caffe

.pth文件------> onnx------->caffe model
（1）pytorch训练的模型转onnx

python convert_mobilenext_onnx.py

（2）将生成的onnx进行简化合并

python -m onnxsim mobilenext_0311.onnx mobilenext_0311_sim.onnx

要安装onnx-simplifier

pip3 install onnx-simplifier
python3 -m onnxsim input_onnx_model output_onnx_model

关于简化onnx，具体参考https://blog.csdn.net/weixin_32642653/article/details/112570329
（3）onnx转caffe模型

python convertCaffe.py mobilenext_0311_sim.onnx mobilenext_0311_sim.prototxt mobilenext_0311_sim.caffemodel

注：转caffe模型是不支持模型中有relu6操作的，要转成relu操作才可以。