###実現したいこと
plot_modelを使ってモデルの可視化を行いたい。
###現状
以下に記述しているコードで学習したモデルの可視化を行うため、plot_model()で出力しようとしたところ、ImportErrorが出ました。
調べてみたところ、pydotの開発が止まっており、python3.5以降だと動作しないとのことでしたので、代わりにpydotplusをインストールし、再度実行してみました。
しかし、同様にplot_model()の部分でImportErrorが発生しております。
pydotplusにパスが通ってることは確認できているのですが、解決法が分かりません。よろしくお願いいたします。
###環境
Python3.6
Tensorflow-gpu 1.9.0
Windows10
python
1import tensorflow as tf 2from tensorflow import keras as K 3from tensorflow.keras.layers import Dense, Activation, Conv2D 4from tensorflow.keras.models import Sequential 5from tensorflow.keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array 6from tensorflow.keras import callbacks 7from tensorflow.keras.callbacks import CSVLogger 8import pydotplus as pydot 9from keras.utils.vis_utils import plot_model 10 11#モデルは省略 12# (x_train, y_trainはNumpy行列の学習データ) 13model.fit_generator( 14 train_data_generator, 15 validation_data=(test_x, test_y), 16 steps_per_epoch=N_TRAIN_DATA // BATCH_SIZE, 17 epochs=50, 18 callbacks=callbacks 19) 20 21pred = model.predict(test_x) 22 23plot_path = os.path.join(model_dir, "model.png") 24plot_model(model, to_file=plot_path, show_shapes=True)
error
1Traceback (most recent call last): 2 File "C:/.../CNN_SR.py", line 121, in <module> 3 plot_model(model, to_file=plot_path, show_shapes=True) 4 File "C:...\lib\site-packages\keras\utils\vis_utils.py", line 133, in plot_model 5 dot = model_to_dot(model, show_shapes, show_layer_names, rankdir) 6 File "C:...\lib\site-packages\keras\utils\vis_utils.py", line 55, in model_to_dot 7 _check_pydot() 8 File "C:...\lib\site-packages\keras\utils\vis_utils.py", line 20, in _check_pydot 9 'Failed to import `pydot`. ' 10ImportError: Failed to import `pydot`. Please install `pydot`. For example with `pip install pydot`.
###pip listの出力
>pip list Package Version ------------------- ------------- absl-py 0.2.2 astor 0.7.1 cycler 0.10.0 gast 0.2.0 grpcio 1.13.0 h5py 2.8.0 Keras 2.2.0 Keras-Applications 1.0.2 Keras-Preprocessing 1.0.1 kiwisolver 1.0.1 llvmlite 0.24.0 Markdown 2.6.11 matplotlib 2.2.2 numba 0.39.0 numpy 1.14.5 opencv-python 3.4.2+contrib pandas 0.23.3 Pillow 5.2.0 pip 10.0.1 protobuf 3.6.0 pydotplus 2.0.2 pyparsing 2.2.0 python-dateutil 2.7.3 pytz 2018.5 PyYAML 3.13 scipy 1.1.0 setuptools 39.1.0 six 1.11.0 tensorboard 1.9.0 tensorflow-gpu 1.9.0 termcolor 1.1.0 Werkzeug 0.14.1 wheel 0.31.1 You are using pip version 10.0.1, however version 18.0 is available. You should consider upgrading via the 'python -m pip install --upgrade pip' command.
回答2件
あなたの回答
tips
プレビュー