[BUG] cannot import name ‘quantize_qat‘ from ‘onnxruntime.quantization‘

前言

进行onnx模型量化出现了这个错误

解决：

卸载重装onnx和onnxruntime，会自动安装最新，由于版本更新取消了quantize_qat函数

改为以下代码：

from onnxruntime.quantization import quantize_dynamic, QuantType,quantize_static

model_fp32 = 'yolov5s.onnx'
model_quant = 'yolov5s_dyn.onnx'
quantized_model = quantize_dynamic(model_fp32, model_quant)

或者使用静态量化见：

模型量化（3）：ONNX 模型的静态量化和动态量化 - 飞桨AI Studio星河社区