Post

[Deep Learning] 딥러닝 모델 구조 시각화

keras를 활용한 딥러닝 모델 구조 시각화.

✅ 딥러닝 모델 구조 시각화

  • keras에는 모델 구조를 이미지로 시각화할 수 있는 Tool이 존재
  • model.summary()로 간단하게 확인할 수 있지만, 표로 더 직관적으로 확인할 수 있다.
  • from tensorflow.keras.utils import plot_model으로 불러와서 plot_model(model)으로 확인할 수 있다.
  • plot_model의 파라미터는 keras Document에서 확인하고 사용하면 된다.
  • 기본적으로 show_shapes=True 하는게 보기는 좋은 듯 하다.

✅ 코드 예시

1
2
3
from tensorflow.keras.utils import plot_model

plot_model(model, show_shapes=True)

image

This post is licensed under CC BY 4.0 by the author.

[Deep Learning] Keras Locally Connection(concatenate, add) 구현

[Deep Learning] 딥러닝 모델 학습 과정 시각화