Ошибка при вызове обученной модели keras

Question

Пытаюсь вызвать обученную модель Keras:

import keras

new_model = keras.models.load_model("/Users/alex/Desktop/model.h5")

Выдает ошибку:

---------------------------------------------------------------------------
OSError                                   Traceback (most recent call last)
<ipython-input-22-7f061beafbf6> in <module>()
      1 import keras
      2 
----> 3 new_model = keras.models.load_model("/Users/alex/Desktop/model.h5")

1 frames
/usr/local/lib/python3.6/dist-packages/tensorflow/python/saved_model/loader_impl.py in parse_saved_model(export_dir)
    111                   (export_dir,
    112                    constants.SAVED_MODEL_FILENAME_PBTXT,
--> 113                    constants.SAVED_MODEL_FILENAME_PB))
    114 
    115 

OSError: SavedModel file does not exist at: /Users/alex/Desktop/model.h5/{saved_model.pbtxt|saved_model.pb}

Answer 1

Спасибо V-Mor за помощь. Заработало по его совету:

в функции save используйте доп параметр save_format='h5'. То есть model.save('Ваш путь', save_format='h5') Вот так в общем: model.save_weights(model_file, save_format='h5')

Answer 2

Чтобы использовать model.load(), которая загружает всю модель целиком, а не только веса, сохранять нужно тоже ВСЮ модель. А делается это с помощью model.save().

Сохраняйте так:

model.save(model_file, save_format='h5')

как гласит документация, по умолчанию модели сохраняются не в формате h5, а в формате tf. Следовательно, необходимо явно указать формат сохранения.

БЛОГ НА HUSL

Ошибка при вызове обученной модели keras

Ответы (2 шт):