Es ist nicht möglich, Ausgabetensoren eines benutzerdefinierten Modells mit drei Ausgängen abzurufen

• Habe einen Ausgabetensor für ein benutzerdefiniertes Modell mit drei Ausgängen:

  from openvino.runtime import Core
  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 = infer_request.get_output_tensor()
  print(output)

• Erhaltener Fehler:

  RuntimeError: get_output_tensor() must be called on a function with exactly one parameter.

Die ov::InferRequest::get_output_tensor Methode ohne Argumente kann für Modelle mit nur einer Ausgabe verwendet werden.

1. Verwenden Sie ov::InferRequest::get_output_tensor Methode mit argument (index: int) für ein Modell, das mehr als eine Ausgabe hat.

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

2. Verwenden Sie das data-Attribut des Tensor-Objekts, um auf die ausgegebenen Tensordaten für die Inferenzergebnisse zuzugreifen.

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