1. keras模型可視化函數
keras.utils.vis_utils模塊提供了可視化Keras模型的函數plot_model,可將模型summary信息以圖片形式輸出。
使用方式如下:
from keras.utils import plot_model
plot_model(model, to_file='model.png')
plot_model 有 4 個可選參數:
- show_shapes (默認為 False) 控制是否在圖中輸出各層的尺寸。
- show_layer_names (默認為 True) 控制是否在圖中顯示每一層的名字。
- expand_dim(默認為 False)控制是否將嵌套模型擴展為圖形中的聚類。
- dpi(默認為 96)控制圖像 dpi
2. 問題解決
plot_model函數依賴 pydot-ng 和 graphviz,若運行出現錯誤,則需要安裝這兩個包:
-
安裝graphviz:Linux系統中可直接用命令行安裝,windows系統下需要下載該二進制安裝包手動安裝,官網下載地址為:下載地址。
-
安裝好之后需要配置環境變量:將graphviz安裝目錄下的bin文件路徑夾添加到Path環境變量中。
-
驗證graphviz是否安裝成功: 進入windows命令行界面,輸入dot -version,然后按回車,如果顯示graphviz的相關版本信息,則安裝配置成功。
-
安裝pydot-ng: 直接在命令行中輸入pip install pydot-ng即可安裝。
【注意】:需要注意的是pydot與pydot-ng的區別,pydot是對應python2的舊版本,python-ng是兼容python2與python3的新版本,之前看博客安裝pydot,運行一直不成功,總是報錯:找不到graphviz,后來才發出現是pydot的問題,我這里使用的是python3,故需要安裝pydot-ng。 -
由於plot_model函數中默認調用的是pydot,安裝pydot-ng后運行代碼仍然會提示找不到pydot,這時需要在源碼...\keras\utils\vis_utils.py模塊中做如下修改, 確認修改,就能正常運行了。
import pydot
修改為
import pydot-ng as pydot