如何将检查点转换为Keras .h5模型？

Question

我有一个保存检查点的tensorflow模型，但是我需要加载权重并保存Kereas .h5模型。我该怎么办？

Answer 1

我假设您需要将以前的检查点转换为.h5

给定一个已经训练好的模型，您想要加载其权重并另存为 .h5 。我假设您已将其另存为 .model 文件。可以说它叫做 first.model

在脚本中，您将要使用load_model，将检查点加载为

model = load_model('first.model')

那么您只需要使用

model.save('goal.h5')

另存为.h5文件。

为将来参考，您可以通过将检查点另存为.h5来避免此转换过程：

使用检查点功能时，您可以选择另存为 .model .h5 或 .hdf5 。该行可能看起来像这样：

checkpoint = ModelCheckpoint("**FILE_NAME_HERE**.model",monitor='val_loss',verbose=1,mode='min',save_best_only=True,save_weights_only=False,period=1)

这就是将检查点另存为.model的方法，但又可以将其另存为h5：

checkpoint = ModelCheckpoint("**FILE_NAME_HERE**.h5",monitor='val_loss',verbose=1,mode='min',save_best_only=True,save_weights_only=False,period=1)