Keras加载模型输入变化

Question

您是否知道在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")

非常感谢， 蒂娜

Answer 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)