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

コメント
コメントを投稿