Convert the model trained by pytorch to caffe

.pth file------>onnx------->caffe model
(1) Convert the model trained by pytorch to onnx

python convert_mobilenext_onnx.py

(2) Simplify and merge the generated onnx

python -m onnxsim mobilenext_0311.onnx mobilenext_0311_sim.onnx

To install onnx-simplifier

pip3 install onnx-simplifier
python3 -m onnxsim input_onnx_model output_onnx_model

About simplifying onnx, please refer to https://blog.csdn.net/weixin_32642653/article/details/112570329
(3) Onnx to caffe model

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

Note: Converting to caffe model does not support relu6 operation in the model, it can only be converted to relu operation.