windows10中keras模型可視化-plot_model

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包,若直接運行會報錯,則需要安裝這兩個包:

  1. 安裝graphviz:Linux系統中可直接用命令行安裝,windows系統下需要下載該二進制安裝包手動安裝,官網下載地址爲:下載地址

  2. 安裝好之後需要配置環境變量:將graphviz安裝目錄下的bin文件路徑夾添加到Path環境變量中。

  3. 驗證graphviz是否安裝成功: 進入windows命令行界面,輸入dot -version,然後按回車,如果顯示graphviz的相關版本信息,則安裝配置成功。

  4. 安裝pydot-ng: 直接在命令行中輸入pip install pydot-ng即可安裝。
    【注意】:需要注意的是pydot與pydot-ng的區別,pydot是對應python2的舊版本,python-ng是兼容python2與python3的新版本,之前看博客安裝pydot,運行一直不成功,總是報錯:找不到graphviz,後來才發出現是pydot的問題,我這裏使用的是python3,故需要安裝pydot-ng。

  5. 由於plot_model函數中默認調用的是pydot,安裝pydot-ng後運行代碼仍然會提示找不到pydot,這時需要在源碼…\keras\utils\vis_utils.py模塊中做如下修改, 確認修改,就能正常運行了。

    import pydot

    修改爲
    import pydot-ng as pydot
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章