获取具有三个输出的自定义模型的输出张量，运行时错误是怎么回事？

答案对人有帮助，有参考价值 0

• 对具有多个输出的模型使用带参数（索引：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)
  
• 使用 Tensor 对象的 data 属性访问推理结果的输出张量数据。output_buffer1 = output2.data
  output_buffer2 = output2.data
  output_buffer3 = output3.data
  print(output_buffer1)
  print(output_buffer2)
  print(output_buffer3)
  

答案对人有帮助，有参考价值 0

在 OpenVINO 中，get_output_tensor() 方法用于获取模型的输出张量。然而，根据错误信息，get_output_tensor() 方法需要一个参数来指定要获取哪个输出张量，特别是在模型有多个输出的情况下。

在你的代码中，模型有三个输出，因此你需要指定要获取哪个输出张量。get_output_tensor() 方法需要一个索引或输出名称作为参数。

解决方法

1. 
   

   使用索引获取输出张量：
   如果你的模型有三个输出，你可以通过索引来获取每个输出张量。索引从 0 开始。

   
   

   output_0 = infer_request.get_output_tensor(0)
   
   output_1 = infer_request.get_output_tensor(1)
   
   output_2 = infer_request.get_output_tensor(2)
   
   print(output_0)
   
   print(output_1)
   
   print(output_2)

   
   


2. 
   

   使用输出名称获取输出张量：
   如果你知道每个输出的名称，你可以通过名称来获取输出张量。

   
   

   output_names = compiled_model.outputs
   
   output_0 = infer_request.get_output_tensor(output_names[0])
   
   output_1 = infer_request.get_output_tensor(output_names[1])
   
   output_2 = infer_request.get_output_tensor(output_names[2])
   
   print(output_0)
   
   print(output_1)
   
   print(output_2)

   
   

完整代码示例

from openvino.runtime import Core



# 初始化 OpenVINO 核心对象

core = Core()



# 读取模型

model = core.read_model(model="model.xml")



# 编译模型

compiled_model = core.compile_model(model, "CPU")



# 创建推理请求

infer_request = compiled_model.create_infer_request()



# 异步推理

infer_request.start_async()

infer_request.wait()



# 获取输出张量

output_0 = infer_request.get_output_tensor(0)

output_1 = infer_request.get_output_tensor(1)

output_2 = infer_request.get_output_tensor(2)



# 打印输出张量

print(output_0)

print(output_1)

print(output_2)

总结

get_output_tensor() 方法需要一个参数来指定要获取哪个输出张量。你可以通过索引或输出名称来指定。确保在调用该方法时传递正确的参数，以避免运行时错误。