Paso 1: Convierta el modelo Dei 5 a Modelo guardado
- Instalar dependencias:
cd \deployment_tools\model_optimizer\install_prerequisites
install_prerequisites_tf2.bat - Cargue el modelo con TensorFlow* 2 y pruébalo en el formato SavedModel.
import tensorflow as tf
model = tf.keras.models.load_model('model.h5')
tf.saved_model.save(model,'model') - Opcional: Modelo con una capa personalizada CustomLayer 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')
Paso 2: Convertir el formato guardado del modelo a representación intermedia
- Cambie el directorio a /deployment_tools/model_optimizer
- Ejecute el script mo_tf.py con una ruta de acceso al directorio SavedModel y un directorio de resultados que se puede generar:
python mo_tf.py --saved_model_dir --output_dir --input_shape