Étapes de conversion du modèle Keras H5 en format IR.
Impossible de déterminer les étapes à suivre pour convertir le modèle Keras* H5 en format IR.
Étape 1 : convertissez le modèle Keras H5 en format de modèle enregistré
- Installer les dépendances :
cd \deployment_tools\model_optimizer\install_prerequisites
install_prerequisites_tf2.bat - 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') - 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
- Modifier le répertoire sur /deployment_tools/model_optimizer
- 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