怎么把.h5文件转换成.pb文件?

时间:2020-02-11 10:46:24

标签: python tensorflow keras tensorflow-serving

如何将.h5重量文件转换为.pb文件?我训练了keras预训练模型,并将文件另存为something.h5。如何将其转换为.pb文件以用于张量流服务?

P.S:请不要提出这个问题,我找不到在线解决方案。请务必提及您使问题降级的原因。否则,请帮助解决问题。

1 个答案:

答案 0 :(得分:2)

import tensorflow as tf
model = tf.keras.models.load_model(keras_model_path)
tf.saved_model.save(model, saved_model_path)

宾果