ขั้นตอนที่ 1: แปลง รุ่น Keras H5 เป็นรูปแบบ SavedModel
- การ ขึ้นต่อกันในการติดตั้ง:
cd \deployment_tools\model_optimizer\install_prerequisites
install_prerequisites_tf2.bat - โหลด รุ่นโดยใช้ TensorFlow* 2 และ ทําให้เป็นอนุกรม ในรูปแบบ SavedModel
import tensorflow as tf
model = tf.keras.models.load_model('model.h5')
tf.saved_model.save(model,'model') - ไม่บังคับ: รุ่นที่มี CustomLayer แบบเลเยอร์แบบกําหนดเองจาก 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')
ขั้นตอนที่ 2: แปลง รูปแบบ SavedModel เป็นการแสดงระดับกลาง
- เปลี่ยน ไดเรกทอรีเป็น /deployment_tools/model_optimizer
- เรียกใช้ สคริปต์ mo_tf.py ด้วยพาธไปยังไดเรกทอรี SavedModel และไดเรกทอรีผลลัพธ์ที่สามารถเขียนได้:
python mo_tf.py --saved_model_dir --output_dir --input_shape