■ 학습 모델 아키텍처 이미지 저장하기

----------------------------------------------------------------------------------------------------


model.data


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

----------------------------------------------------------------------------------------------------

Posted by 사용자 icodebroker

댓글을 달아 주세요