如何将.h5重量文件转换为.pb文件?我训练了keras预训练模型,并将文件另存为something.h5。如何将其转换为.pb文件以用于张量流服务?
P.S:请不要提出这个问题,我找不到在线解决方案。请务必提及您使问题降级的原因。否则,请帮助解决问题。
答案 0 :(得分:2)
import tensorflow as tf
model = tf.keras.models.load_model(keras_model_path)
tf.saved_model.save(model, saved_model_path)
宾果