.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操作才可以。