导入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模型。
父主题: 解析框架模型