.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操作にのみ変換できます。