Méthode correcte pour obtenir le tenseur de sortie pour un modèle qui a plusieurs sorties
- Vous avez un tenseur de sortie pour un modèle personnalisé qui a trois sorties :
à partir du cœur d’importation openvino.runtime
cœur = cœur()
model = core.read_model(model="model.xml »)
compiled_model = core.compile_model(modèle, « CPU »)
infer_request = compiled_model.create_infer_request()
infer_request.start_async()
infer_request.wait()
output = infer_request.get_output_tensor()
impression (sortie) - Erreur reçue :
RuntimeError : get_output_tensor() doit être appelé sur une fonction avec exactement un paramètre.
La méthode ov :: InferRequest ::get_output_tensor sans arguments peut être utilisée pour un modèle avec une seule sortie.
- Utilisez la méthode ov :: InferRequest ::get_output_tensor avec argument (index : int) pour le modèle qui a plusieurs sorties.
output1 = infer_request.get_output_tensor(0)
output2 = infer_request.get_output_tensor(1)
output3 = infer_request.get_output_tensor(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)