■ 학습 모델 아키텍처 이미지를 저장하는 방법을 보여준다.
▶ 예제 코드 (PY)
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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() |
