小記
最近準備在寒假結束前徹底解決掉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第一個區塊所對應的圖片