[Keras Visualization] plot_model 사용법

Keras Documentation을 보면 Visualization 기능을 제공하는 것을 확인할 수 있다.

https://keras.io/visualization/

그 내용을 살펴보면 다음과 같다.

The keras.utils.vis_utils module provides utility functions to plot a Keras model (using graphviz).

This will plot a graph of the model and save it to a file:

graphviz를 이용해서 model의 graph를 plot형태로 파일에 저장을 해준다고한다.

먼저 plot_model을 위해서는 graphviz를 먼저 설치해야겠다.

graphviz의 공식 홈페이지는다음과 같다.

https://www.graphviz.org/

먼저 단순히 graphviz만 설치한 결과 이미지 파일로 저장이 안되는 문제가 발생했다.

graphviz 설치전 파일을 이미지 파일로 저장하기위한 라이브러리들을 설치해주자.

sudo apt install -y libgd-dev 

sudo apt install -y fontconfig

sudo apt install -y libcairo2-dev

sudo apt install -y libpango1.0-dev

sudo apt install -y libgts-dev

graphviz를 설치 방법은 2가지가 존재한다.

먼저 간단하게 pip를 이용한 방법과 source 코드를 다운받아서 직접 컴파일 하는 방법이다.

pip를 이용한 설치 방법

pip install graphviz

다음은 직접 소스코드를 다운받아 설치하는 방법이다.

wget https://graphviz.gitlab.io/pub/graphviz/stable/SOURCES/graphviz.tar.gz

tar -xvf graphviz.tar.gz

cd graphviz

./configure

make

sudo make install

설치가 완료되면 이제 저장하고 싶은 model을 입력만 해주면 된다.

from keras.utils import plot_model plot_model(model, to_file='model.png')

해당 폴더를 확인해보면 model.png 파일이 생긴 것을 확인할 수 있다.