导入ONNX模型

  1. 新建builder,由builder创建network。
  2. 创建OnnxModelParser对象。
  3. 调用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模型。