[PYTHON/KERAS] 학습 모델 아키텍처 이미지 저장하기

■ 학습 모델 아키텍처 이미지를 저장하는 방법을 보여준다.

 

▶ 예제 코드 (PY)

import keras.models as models
import keras.utils as utils
import matplotlib.image as img
import matplotlib.pyplot as pp

# 모델 데이터를 로드한다.
model = models.load_model("model.data")

# 모델 아키텍처 이미지를 저장한다.
utils.vis_utils.plot_model(model, to_file = "model.png", show_shapes = True, show_layer_names = True)

# 모델 아키텍처 이미지 파일을 로드한다.
imageNDArray = img.imread("model.png")

pp.imshow(imageNDArray)

pp.show()