как преобразовать файл .h5 в файл .pb? - PullRequest
0 голосов
/ 11 февраля 2020

Как мне преобразовать файл веса .h5 в файл .pb? Я подготовил керас предварительно обученной модели и сохранил файл как-то .h5. Как мне преобразовать его в файл .pb, чтобы использовать его для обслуживания tenorflow?

PS: Не унижайте вопрос, я не смог найти никакого решения в Интернете. Укажите причину, по которой вы унизили вопрос. В противном случае помогите с решением вопроса.

1 Ответ

1 голос
/ 12 февраля 2020
import tensorflow as tf
model = tf.keras.models.load_model(keras_model_path)
tf.saved_model.save(model, saved_model_path)

Корзина go.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...