本来以为安装完onnx可以直接使用onnxsim,发现还需要安装onnxsim
- step 1 安装onnxsim
pip install onnx-simplifier
导出的onnx模型包含许多冗余的维度,这是不支持ncnn的,所以需要进行去掉冗余的维度。
- step 2 使用命令导出sim.onnx模型
python -m onnxsim BiSeNet.onnx BiSeNet-sim.onnx
或者
加载onnx文件,simplify处理后重新保存,代码如下
from onnxsim import simplify
onnx_model = onnx.load(output_path

在将ONNX模型转换为NCNN的过程中,首先需要安装onnxsim以去除模型中的冗余维度。通过使用onnxsim简化模型后,可以使用onnx2ncnn工具将模型转换为NCNN所需的parm和bin文件。编译ncnn并参考相关教程完成整个流程。
订阅专栏 解锁全文

6370

被折叠的 条评论
为什么被折叠?



