本来以为安装完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