tensorflow - “模型”对象没有属性 'load_model' keras

Question

我正在尝试加载一个保存它的模型:model.save('myModel.h5')
模型定义如下:

self.model = VGGFace(input_tensor=input_tensor, include_top=True)

for layer in self.model.layers:
    layer.trainable = False

self.model.get_layer('fc7').trainable = True
last_layer = self.model.get_layer('fc7').output
out = BatchNormalization()(last_layer)
out = Dense(self.n_outputs, activation='softmax', name='fc8')(out)
self.model = Model(input=self.model.input, output=out)

当我尝试加载 myModel.h5与 model.load_model('myModel.h5')它向我抛出以下错误:

AttributeError: 'Model' object has no attribute 'load_model'

我猜是因为我没有和 Sequential 一起工作楷模。

那么我如何加载我的模型？自 model.save('myModel.h5')似乎工作。

谢谢!!!!

Answer 1

load_model()确实不是模型对象的属性。 load_model()是一个从 keras.models 导入的函数，它接受一个文件名并返回一个模型对象。

你应该像这样使用它:

from keras.models import load_model

model = load_model(path_to_model)

You can then use keras.models.load_model(filepath) to reinstantiate your model. load_model will also take care of compiling the model using the saved training configuration (unless the model was never compiled in the first place). from source