【如何检查 ONNX 模型是否正确？】onnx.checker.check_model 用法详解

以下是对 onnx.checker.check_model 函数文档的翻译：

onnx.checker.check_model

检查模型的一致性，即模型在结构、格式和配置方面的正确性和完整性。

如果模型的 ir_version 设置不正确或高于检查器的 ir_version，或者模型在 metadata_props 中有重复键，将引发异常。

如果 IR version >= 3，模型必须指定 opset_import。如果 IR version < 3，模型不能指定 opset_import。

参数：

• model：要检查的模型。如果模型是一个路径，函数会首先检查模型路径。如果模型的字节大小超过2GB，应使用模型路径来调用该函数。
• full_check：如果为 True，函数还会运行形状推断检查。
• skip_opset_compatibility_check：如果为 True，函数将跳过算子集兼容性检查。
• check_custom_domain：如果为 True，函数将检查所有域。否则，仅检查内置域。

示例代码

import onnx

# 检查 onnx 计算图
onnx_model = onnx.load("/path/to/yourmodel.onnx")
# 检查模型
try:
    onnx.checker.check_model(onnx_model)
    print("Model check passed.")
except Exception as e:
    print(f"Model check failed: {e}")

如果没问题，就会打印出 “Model check passed.”

参考内容：
[1] onnx.checker.check_model 相关文档