Onnx 转Ncnn

Onnx 转Ncnn

算法工程师给了onnx,需要转成ncnn才能用到安卓上去,步骤如下

简化onnx

算法给了.onnx后缀的文件,100多兆,太大,第一步,先简化:

  1. conda env list 查看conda环境(前提是之前已经配了)
  2. conda activate pytorch 激活名字叫pytorch的环境
  3. 把自己的onnx模型拷贝到某目录,然后cd进入某目录
  4. pip install onnx-simplifier 安装最新版本的onnx简化插件
  5. python -m onnxsim 20230406_e263.onnx 20230406_e263_sim.onnx --skip-optimization eliminate_duplicate_initializer 执行指令简化
    Onnx 转Ncnn_第1张图片
    6.下载编译好的onnx2ncnn.exe工具,地址:https://github.com/Tencent/ncnn/releases
    Onnx 转Ncnn_第2张图片
    7、执行指令 onnx2ncnn.exe 20230406_e263_sim.onnx yolov5s.param yolov5s.bin
    Onnx 转Ncnn_第3张图片

8,把这两个文件放入项目,然后再按照下面这个链接处理就好
https://blog.csdn.net/qq_45077760/article/details/129802464

你可能感兴趣的:(python,开发语言)