跳转至主要内容
支持知识库

无法获取具有三个输出的自定义模型的输出张量

内容类型: 错误讯息   |   文章 ID: 000090966   |   上次审核日期: 2023 年 11 月 13 日

环境

OpenVINO™ 2022.1.0

说明

  • 获取具有三个输出的自定义模型的输出张量:

    从 openvino.runtime 导入内核
    内核 = 内核()
    model = core.read_model(model=“model.xml”)
    compiled_model = core.compile_model(型号,“CPU”)
    infer_request = compiled_model.create_infer_request()
    infer_request.start_async()
    infer_request.wait()
    output = infer_request.get_output_tensor()
    打印(输出)

  • 收到的错误:

    运行时错误:必须对只有一个参数的函数调用 get_output_tensor()。

解决方法

不带参数的 ov::InferRequest::get_output_tensor 方法可用于只有一个输出的模型。

  1. 对具有多个输出的模型使用带参数(索引:int)的 ov::InferRequest::get_output_tensor 方法。

    output1 = infer_request.get_output_tensor(0)
    output2 = infer_request.get_output_tensor(1)
    output3 = infer_request.get_output_tensor(2)

  2. 使用 Tensor 对象的 data 属性访问推理结果的输出张量数据。

    output_buffer1 = output2.data
    output_buffer2 = output2.data
    output_buffer3 = output3.data
    print(output_buffer1)
    print(output_buffer2)
    print(output_buffer3)

相关产品

本文适用于 3 产品。
英特尔® 至强融核™ 处理器软件 OpenVINO™ 工具包 性能函数库

免责声明

本页面上的内容是原始英文内容的人工翻译与计算机翻译的组合。我们提供此内容是为了您的便利并且仅供参考,未必完整或准确。如果本页面的英文版与翻译版之间存在任何冲突,应以英文版为准。 查看此页面的英语版本。