onnx转ncnn

本来以为安装完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)  # load onnx model
model_simp, check = simplify

你可能感兴趣的:(onnx,ncnn)