Visualizing Keras Model: ImportError 에러 해결방법

Reference : http://www.codesofinterest.com/2017/02/visualizing-model-structures-in-keras.html

윈도우 + 아나콘다 + Keras 사용시 모델을 Visualize 할 때 발생하는 에러에 대한 해결책.

 

InceptionV3 Model

 

원래 간단하게는

pip install pydot
pip install graphviz

를 설치한 뒤에

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

하면되나, ImportError가 발생해버린다.

이를 해결하기 위해서는, pydot은 pip로 설치하고, graphviz는 공식 사이트(https://www.graphviz.org/ )에서 윈도우 설치판을 다운받도록 하고 설치한다. (설치경로 : C:/Program Files (x86)/Graphviz2.38/bin/ )

그리고 이 주소를 환경변수 PATH에 추가한 뒤,

python 소스에는

import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'

이렇게 넣어주면 문제없이 출력된다.

0 Shares:
Leave a Reply

Your email address will not be published. Required fields are marked *

This site uses Akismet to reduce spam. Learn how your comment data is processed.