ID de l'article: 000090966 Type de contenu: Messages d'erreur Dernière révision: 13/11/2023

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

Environnement

OpenVINO™ 2022.1.0

BUILT IN - ARTICLE INTRO SECOND COMPONENT
En bref

Méthode correcte pour obtenir le tenseur de sortie pour un modèle qui a plusieurs sorties

Descriptif
  • 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.

Résolution

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

  1. 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)

  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)

Produits associés

Cet article concerne 1 produits

Le contenu de cette page est une combinaison de traduction humaine et informatique du contenu original en anglais. Ce contenu vous est fourni pour votre commodité et à titre informatif seulement et ne saurait être totalement exact ou complet. En cas de contradiction entre la version anglaise de cette page et la traduction, c'est la version anglaise qui prévaut. Afficher la version anglaise de cette page.