Keras.utils.vis_utils.plot_model 繪制神經網絡結構示意圖


keras的內置函數keras.utils.vis_utils.plot_model可以直接將搭建的神經網絡用流程圖表示出來

 

該函數原來為keras.utils.visualize_util.plot,在2017年3月1日的更新中作了修改

visualize_util改為vis_utils;plot改為plot_model;

使用原函數名會報錯import error

 

使用該函數需要額外的函數庫和軟件,以下說明引自博客:

https://www.jianshu.com/p/3a97c0a43beb

1. (使用pip install)安裝pydot和graphviz兩個函數庫

2. 安裝graphviz軟件:

https://graphviz.gitlab.io/_pages/Download/Download_windows.html

3. 將路徑加入系統變量,從我的電腦-屬性-高級系統設置-環境變量 進入

(圖片為引用,根據安裝路徑調整)

4. 檢查graphviz是否安裝成功,進入windows命令行,輸入命令:dot -version,若成功顯示版本信息,則認為安裝和添加路徑成功

5. 引用函數需要以下語句

from keras.utils.vis_utils import plot_model  

import os

os.environ["PATH"] += os.pathsep +'D:/Program Files (x86)/Graphviz2.38/bin/' # 后一段路徑改為自己安裝的路徑

6. 繪圖語句如下:

plot_model(model,to_file='name.png',show_shapes=True)

結果如圖所示:

 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM