2017-05-15 4 views
4

테스트 용 다른 프레임 워크에서 덤프 가중치 및 모델 아키텍처를 사용하려고합니다. model.get_config() 모델의 구성을 제공 할 수 있습니다 알아요. model.to_json은 모델의 표현을 JSON 문자열로 반환합니다. 그러나 표현에는 가중치가 포함되지 않으며 아키텍처 만 포함됩니다. 그리고 model.save_weights (filepath) : 모델의 가중치를 HDF5 파일로 저장합니다. model.save_weights (filepath)는 모델의 가중치를 HDF5 파일로 저장합니다. json 파일에 아키텍처뿐만 아니라 가중치도 저장하려고합니다.keras를 사용하여 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())