OnnxParser 是 TensorRT 中用于将 ONNX 模型转换为 TensorRT 网络定义的核心解析器。以下从功能到调试全面解析其工作原理及实践方法:
一、核心功能
功能 | 说明 |
---|---|
模型加载 | 读取 .onnx 文件,验证模型结构完整性 |
算子映射 | 将 ONNX 算子转换为 TensorRT 层(如 Conv → IConvolutionLayer) |
权重提取 | 解析模型中的常量张量(如卷积核权重、偏置项) |
网络构建 | 生成 INetworkDefinition 对象,供后续优化 |
二、核心类与方法
1. 类结构
-