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)