导入ONNX模型
- 新建builder,由builder创建network。
- 创建OnnxModelParser对象。
- 调用parse()方法。
C++示例参考如下:
Builder* builder = Builder::CreateInferBuilder("Ascend310P3"); // 构造ONNX解析器 auto network = builder->CreateNetwork(); // 创建network std::string modelPath("./resnet50.onnx"); OnnxModelParser parser; bool ret = parser.Parse(network, modelPath.c_str()); // 解析ONNX模型生成network

当前ONNX parser解析支持resnet50,resnet18, resnet101模型。
父主题: 解析框架模型