1 在windows上转换onnx模型为ncnn的param,bin格式
在Windows系统上可以直接使用ncnn在windows上的发布包里面的onnx2ncnn.exe
程序对onnx模型进行转换,我们可以在ncnn的发布页https://github.com/Tencent/ncnn/releases 直接下载ncnn在windows上的发布包,如下图所示
这里以上图中的ncnn-20230223-windows-vs2019-shared.zip为例,解压缩包之后进入到x64/bin文件夹中就可以看到onnx2ncnn.exe
,如下图
然后我们使用命令行工具cd到该目录下,假设需要转换的onnx模型为mobile.onnx,这时候将mobile.onnx模型拷贝到该目录下,使用以下命令行完成模型转换
./onnx2ncnn.exe ./mobile.onnx ./mobile.param ./mobile.bin
如果没有错误,执行成功之后会在该目录下生成mobile.param和mobile.bin两个文件,这个就是ncnn所需要的模型和参数文件。
如果在转换过程中出现了以下的错误
Shape not supported yet!
Gather not supported yet!
Unsupported unsqueeze axes !
Unknown data type 0
则需要在在python中使用onnxsim库对模型进行简化然后再转换成ncnn。
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:NCNN – 在windows上转换onnx模型为ncnn的param,bin格式
原文链接:https://www.stubbornhuang.com/2560/
发布于:2023年03月22日 15:58:11
修改于:2023年12月07日 9:50:48
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。
评论
52