pytorch2ncnn学习记录

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