dump weights of cnn in json using keras

后端 未结 1 1087
长发绾君心
长发绾君心 2021-01-03 01:49

I want to use the dumped weights and model architecture in other framework for testing.

I know that:

  • model.get_config() can give the conf
相关标签:
1条回答
  • 2021-01-03 01:58

    Keras does not have any built-in way to export the weights to JSON.

    Solution 1:

    For now you can easily do it by iterating over the weights and saving it to the JSON file.

    weights_list = model.get_weights()
    

    will return a list of all weight tensors in the model, as Numpy arrays.

    Then, all you have to do next is to iterate over this list and write to the file:

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

    Solution 2:

    import json
    weights_list = model.get_weights()
    print json.dumps(weights_list.tolist())
    
    0 讨论(0)
提交回复
热议问题