ONNX 开始
环境
基础
conda create -n onnx python=3.8 -y
conda activate onnx
# ONNX
# https://github.com/onnx/onnx
conda install -c conda-forge onnx -y
python -c "import onnx; print(onnx.__version__)"
import onnx
model = onnx.load("model.onnx")
简化
# ONNX Simplifier
# https://github.com/daquexian/onnx-simplifier
pip install onnx-simplifier
python -m onnxsim -h
import onnxsim
model_simp, check = onnxsim.simplify(model, perform_optimization=False)
assert check, "Simplified ONNX model could not be validated"
使用
给出些 ONNX 模型使用的示例方法。