无法获取具有三个输出的自定义模型的输出张量
内容类型: 错误讯息 | 文章 ID: 000090966 | 上次审核日期: 2023 年 11 月 13 日
从 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 方法可用于只有一个输出的模型。
output1 = infer_request.get_output_tensor(0)
output2 = infer_request.get_output_tensor(1)
output3 = infer_request.get_output_tensor(2)
output_buffer1 = output2.data
output_buffer2 = output2.data
output_buffer3 = output3.data
print(output_buffer1)
print(output_buffer2)
print(output_buffer3)