ONNX版本降级
最近发现了onnx库中其实可以直接修改输出模型的format版本,代码如下:
import onnx
model = onnx.load(r"C:\Users\33132\Desktop\exp14\weights\SourceModelName.onnx")
model.ir_version = 6
onnx.save_model(model, r"C:\Users\33132\Desktop\ObjectModelName.onnx")
上述代码仅需准备好原始ONNX模型文件即可,然后写好路径后运行上述路径即可完成模型降级。