pytorch(.pt)->onnx(.onnx)->ncnn(.param .bin)
1.pytorch的.pt模型转为.onnx
2.使用工具 onnx-simplifier 来简化 onnx
python -m onnxsim origin.onnx origin-sim.onnx
#origin-sim.onnx为生成的简化模型
3.onnx2ncnn(win10系统下)
参考链接:https://zhuanlan.zhihu.com/p/362679768
主要需要编译ncnn生成onnx2ncnn的可执行文件,具体过程参考上面的链接。然后再pycharm终端输入:
<onnx2ncnn root dir>\onnx2ncnn model.onnx model.param model.bin
< onnx2ncnn root dir>是onnx2ncnn.exe所在路径
最终生成model.param和model.bin