Dump Gewichte von CNN in Json mit Keras

Question

Ich möchte die Dumping-Gewichte und Modellarchitektur in anderen Rahmen zum Testen verwenden. Ich weiß, dass model.get_config() die Konfiguration des Modells geben kann. model.to_json gibt eine Repräsentation des Modells als JSON-String zurück. Aber dass die Darstellung nicht die Gewichte enthält, nur die Architektur. Und model.save_weights (Dateipfad): speichert die Gewichtungen des Modells als HDF5-Datei. und model.save_weights (Dateipfad) speichert die Gewichtungen des Modells als HDF5-Datei. Ich möchte die Architektur sowie Gewichte in einer JSON-Datei speichern.

Answer 1

Keras verfügt nicht über eine integrierte Möglichkeit, die Gewichtungen in JSON zu exportieren.

Lösung 1:

Vorerst Sie leicht tun können, über die Gewichte durch Iterieren und in die JSON-Datei zu speichern.

gibt eine Liste aller Gewichtstensoren im Modell als Numpy-Arrays zurück.

Dann alles, was Sie tun müssen, um als nächstes über diese Liste zu durchlaufen und in die Datei schreiben:

for i, weights in enumerate(weights_list): 
    writeJSON(weights) 

Lösung 2:

import json 
weights_list = model.get_weights() 
print json.dumps(weights_list.tolist())