Keras で plot_model() を使って、下記エラーが出力された場合、
"Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work."
原因は、graphviz.exe が見つからないのが原因です。システム環境変数 Path に、graphviz.exe への Path を追加することで解決しました。
Anaconda 4.3.0
Python 3.6
Tensorflow 1.4.0
Keras 2.1.2
https://keras.io/visualization/
"Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work."
原因は、graphviz.exe が見つからないのが原因です。システム環境変数 Path に、graphviz.exe への Path を追加することで解決しました。
動作環境
Windows 10 Pro (64bit)Anaconda 4.3.0
Python 3.6
Tensorflow 1.4.0
Keras 2.1.2
手順
step 1. graphviz をダウンロードします
https://graphviz.gitlab.io/_pages/Download/Download_windows.htmlstep 2. システム環境変数 Path に graphviz.exe の場所を追加します
C:\Program Files (x86)\Graphviz2.38\binstep 3. Neural Network を図として出力します
import keras from keras.models import load_model from keras.utils import plot_model import os model_dir = os.path.join(os.getcwd(), 'models') model_name = 'keras_mnist_trained_model.h5' model_graph_name = 'keras_mnist_trained_model.png' model_path = os.path.join(model_dir, model_name) if os.path.isfile(model_path): model = load_model(model_path) print('Load trained model at %s ' % model_path) model_graph_path = os.path.join(model_dir, model_graph_name) plot_model(model, to_file=model_graph_path, show_shapes=True)
step 4. Coffee break, 昼食をとりましょう
図 時屋のどらやき, Subsurface Scattering
出力例
図 MNIST 学習のネットワーク
出典
Keras - plot_model()https://keras.io/visualization/
HDF5
HDF5, JAXA
コメント
コメントを投稿