как получить сессию Tensorflow только из файла .h5 keras без сессии - PullRequest
1 голос
/ 24 октября 2019

Мотивация этого вопроса заключается в том, что я сохранил модель Keras, используя Matterport's MaskRCNN, и в tf.keras.callbacks.ModelCheckpoint() очень явно установил аргумент save_weights_only в значение False, чтобы была сохранена вся модель (не только веса).

Оказывается, есть ошибка в обратном вызове ModelCheckpoint (), из-за которой иногда не сохраняется полная модель.

Это, очевидно, проблема при загрузке модели после закрытия TF. сеанса, поскольку состояние графика, архитектуры и оптимизатора исчезло, что затрудняет (если не невозможно) перезагрузку сохраненной модели.

Поэтому я спрашиваю, возможно ли каким-либо образом извлечь сеанс TF задним числомтолько из файла весов .h5 после закрытия сеанса (например, из-за сбоя ядра Notebook).

Не так много кода для продолжения, но он есть:

Учитывая .h5 файл, который был сохранен после каждой эпохи обучения модели в Керасе, возможно ли извлечь сессию Графаиз этого файла .h5, и если да, то как?

У меня есть несколько моделей, сохраненных в формате .h5, но я никогда не вызывал tf.get_session () во время сохранения весов моделей в формате h5.

with tf.session() as sess:

как загрузить эту модель с помощью Tensorflow

TF 2.0 делает это несложно, но как решить эту проблему в версии Tensorflow 1.14?

Конечная цель этого состоит в том, чтобымодель, сохраненная с Keras в виде файла .h5, и сделайте вывод о ней в Tensorflow Serving, который, насколько мне известно, нуждается в файле protobuf в формате .pb.

https://medium.com/@pipidog/how-to-convert-your-keras-models-to-tensorflow-e471400b886a

Я пробовал keras_to_tensorflow: https://github.com/amir-abdi/keras_to_tensorflow

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