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

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

model.zip