Convierta el modelo entrenado por pytorch en caffe

Archivo .pth ------> onnx -------> modelo caffe
(1) Convierta el modelo entrenado por pytorch a onnx

python convert_mobilenext_onnx.py

(2) Simplificar y fusionar el onnx generado

python -m onnxsim mobilenext_0311.onnx mobilenext_0311_sim.onnx

Para instalar onnx-simplifier

pip3 install onnx-simplifier
python3 -m onnxsim input_onnx_model output_onnx_model

Con respecto a la simplificación de onnx, consulte 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

Nota: La conversión al modelo caffe no admite la operación relu6 en el modelo, solo se puede convertir a la operación relu.