2017-03-31 6 views
0

Haben Sie eine Idee von einer einfachen Möglichkeit, Eingabebildgröße eines gespeicherten Modells in Keras zu ändern? Zum Beispiel ist die Trainingseingabebildgröße 32x32, aber im Test möchte ich das vollständige Bild 180x180 eingeben. Das Modell wurde gespeichert und bei Test wie die folgenden geladen: 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 geladene Modelleingabe ändern

Vielen Dank, Tina

Antwort

0

Ist das ein vollständig Faltungs Netz? Andernfalls können Sie es nicht mit einer anderen Eingabegröße wiederverwenden, da sich dadurch die Anzahl der Gewichtungen in den nicht konvolutionellen Schichten ändert.

Wenn es in der Tat ein FCN ist, müssen Sie nur die erste und die letzte Zeile im Code ändern, um das Modell zu definieren:

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)