Impossible d’obtenir les tenseurs de sortie du modèle personnalisé qui a trois sorties

• Vous avez un tenseur de sortie pour un modèle personnalisé qui a trois sorties :

  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)

• Erreur reçue :

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

La méthode ov::InferRequest::get_output_tensor sans arguments peut être utilisée pour un modèle avec une seule sortie.

1. Utilisez ov::InferRequest::get_output_tensor méthode avec argument (index: int) pour le modèle ayant plusieurs sorties.

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

2. Utilisez l’attribut data de l’objet Tensor pour accéder aux données du tenseur de sortie pour les résultats de l’inférence.

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