Langkah-langkah untuk mengonversi model Keras H5 ke format IR.
Tidak dapat menentukan langkah-langkah untuk mengonversi model Keras* H5 ke format IR.
Langkah 1: Mengonversi model Keras H5 ke Format SavedModel
- Instal keandalan:
cd \deployment_tools\model_optimizer\install_prerequisites
install_prerequisites_tf2.bat - Muat model menggunakan TensorFlow* 2 dan serialisasi dalam format SavedModel.
import tensorflow as tf
model = tf.keras.models.load_model('model.h5')
tf.saved_model.save(model,'model') - Opsional: Model dengan CustomLayer lapisan kustom dari 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')
Langkah 2: Mengonversi Format SavedModel menjadi Representasi Menengah
- Ubah direktori menjadi /deployment_tools/model_optimizer
- Jalankan skrip mo_tf.py dengan jalur ke direktori SavedModel dan direktori keluaran yang dapat ditulis:
python mo_tf.py --saved_model_dir --output_dir --input_shape
Mengonversi Model TensorFlow 2 dalam Mengonversi Model TensorFlow.
Parameter Konversi Umum dan Kapan Menentukan Bentuk Input dalam Mengonversi Model menjadi Representasi Menengah.