pytorchによってトレーニングされたモデルをcaffeに変換します

.pthファイル------> onnx -------> caffeモデル
(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 tocaffeモデルを参照してください。

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

注:caffeモデルへの変換は、モデルでのrelu6操作をサポートしていません。relu操作にのみ変換できます。

おすすめ

転載: blog.csdn.net/qq_35037684/article/details/114694800