2017-05-15 6 views
4

私は、テストのために他のフレームワークでダンプされたウエイトとモデルアーキテクチャを使いたいと思っています。私はmodel.get_config()がモデルの設定を与えることができます知っている。 model.to_jsonはモデルの表現をJSON文字列として返します。しかし、表現には重みは含まれず、アーキテクチャのみが含まれます。 model.save_weights(filepath):モデルの重みをHDF5ファイルとして保存します。 model.save_weights(filepath)は、モデルの重みをHDF5ファイルとして保存します。 jsonファイルにアーキテクチャと重みを保存します。ケラスを使用してjsonでcnnのダンプウェイトを

答えて

3

Kerasには、重みをJSONにエクスポートする組み込みの方法はありません。

解決方法1:

今のあなたには、重みを反復処理し、JSONファイルに保存することによってそれを簡単に行うことができます。

weights_list = model.get_weights() 

は、Numpy配列としてモデル内のすべての重みテンソルのリストを返します。

次に、あなたが次に何をする必要があるすべては、このリストを反復処理すると、ファイルに書き込むことです。

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

解決方法2:

import json 
weights_list = model.get_weights() 
print json.dumps(weights_list.tolist()) 
関連する問題