ID de l'article: 000088667 Type de contenu: Dépannage Dernière révision: 20/05/2022

Comment convertir le modèle Keras* H5 en format de représentation intermédiaire ?

BUILT IN - ARTICLE INTRO SECOND COMPONENT
En bref

Étapes de conversion du modèle Keras H5 en format IR.

Descriptif

Impossible de déterminer les étapes à suivre pour convertir le modèle Keras* H5 en format IR.

Résolution

Étape 1 : convertissez le modèle Keras H5 en format de modèle enregistré

  1. Installer les dépendances :
    cd \deployment_tools\model_optimizer\install_prerequisites
    install_prerequisites_tf2.bat
  2. Chargez le modèle à l’aide de TensorFlow* 2 et sériez-le au format SavedModel.
    import tensorflow as tf
    model = tf.keras.models.load_model('model.h5')
    tf.saved_model.save(model,'model')
  3. Facultatif : Modèle avec une couche personnalisée CustomLayer à partir de custom_layer.py :
    import tensorflow as tf
    from custom_layer import CustomLayer
    model = tf.keras.models.load_model('model.h5', custom_objects={'CustomLayer': CustomLayer})
    tf.saved_model.save(model,'model')

 

Étape 2 : convertissez le format de modèle enregistré en représentation intermédiaire

  1. Modifier le répertoire sur /deployment_tools/model_optimizer
  2. Exécutez le script mo_tf.py avec un chemin vers le répertoire SavedModel et un répertoire de sortie writable :
    python mo_tf.py --saved_model_dir --output_dir --input_shape

Produits associés

Cet article concerne 2 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.