如何从 .h5 文件加载保存的 keras 权重

问题描述 投票:0回答:1

我正在尝试复制图像分类的示例 - 训练、验证,最后在独立图像样本上进行测试

我(仅)使用以下方法保存了权重:

model.save_weights('file_name.h5')

现在,为了运行预测,我想加载我正在尝试的 saved

.h5
文件:

Model.load_weights('file_name.h5')

文件

file_name.h5
保存在同一根路径中。

但是,我收到错误

TypeError:load_weights() 缺少 1 个必需的位置参数: '文件路径'

我应该怎么做才能装载节省的重量。

keras 页面有这方面的讨论

python-3.x tensorflow keras-2
1个回答
0
投票

只是为了关闭愚蠢问题(无法删除它)(除了typho_s):

这就是我要做的。

  1. 创建模型后:

    model = Sequential()
    model.add(Activation('relu'))
    ...
    ...
    model.add(Dense(1))
    model.add(Activation('sigmoid'))
    
  2. 加载重物:

    模型.load_weights()

© www.soinside.com 2019 - 2024. All rights reserved.