Keras 加载模型输入变化

【问题标题】：Keras loaded model input changeKeras 加载模型输入变化
【发布时间】：2017-04-15 20:58:37
【问题描述】：

您知道在 Keras 中修改已保存模型的输入图像大小的简单方法吗？例如训练输入图像大小为 32x32，但在测试中我想输入 180x180 的完整图像。模型已保存并在测试时加载如下： json_file = open('autoencoder64a.json', 'r') loaded_model_json = json_file.read() json_file.close() loaded_model = model_from_json(loaded_model_json) # load weights into new model loaded_model.load_weights("autoencoder64a.h5")

非常感谢，蒂娜

【问题讨论】：

标签： keras

【解决方案1】：

这是一个全卷积网络吗？否则，您将无法以不同的输入大小重复使用它，因为这会改变非卷积层中的权重数量。

如果确实是FCN，只需要修改定义模型的代码中的第一行和最后一行即可：

    input_layer = Input((180,180))
    #All other layers copied here from your old model, 
    #ending with 'last_layer =...'
    new_model = Model(input_layer, last_layer)

【讨论】：