用圖片存儲和顯示keras模型的結構

小記

最近準備在寒假結束前徹底解決掉keras的使用方法,並且在下學期結束掉tensorflow的使用,然後開始一心一意的把keras和tensorflow的源代碼全部學一遍,這樣才能徹掌握標準的python程序框架的寫法。

keras的可視化

keras本身提供了可視化的庫文件,但是由於keras的升級,已經拋棄了原來的接口和庫,所以名字已經發生了改變,用網上的教程是走不通的。這裏主要有三個庫:

  • pydot
  • pydot_ng
  • graphviz

由於keras新版本不再使用舊的pydot接口,所以查看源代碼後實際使用的是pydot_ng,原來的pydot只在異常處理的時候配合pydotplus使用。另外這裏的graphviz是不能用pip安裝的,安裝完成後是不會識別的,一定一定要用sudo apt-get install graphviz 安裝。然後在程序中使用:


from keras.utils.vis_utils import plot_model
plot_model(Mod, to_file='model1.png',show_shapes=True)

就可以在指定的路徑下生成模型對應的圖片,下面是我resnet第一個區塊所對應的圖片
MODE

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