用角码转储cnn在json中的权重

Question

我希望在其他框架中使用转储权值和模型体系结构进行测试。

我知道：

• model.get_config()可以给出模型的配置。
• model.to_json以JSON字符串的形式返回模型的表示，但表示不包括权重，只包含体系结构
• model.save_weights(filepath)将模型的权重保存为HDF5文件

我想在json文件中保存体系结构和权重。

Answer 1

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