Keras中模型架构可视化

Keras中模型架构可视化

1. 安装相关Python package

    pip install graphviz
    pip install pydot
    pip install pydot-ng

2. 安装graphviz

  graphviz官网

3. 设置graphviz环境变量

注：graphviz使用默认安装位置

  选择“此电脑”，右键“属性”，进入“系统”界面；选择“高级系统设置”，进入“系统属性”界面；单击右下角“环境变量(N)”，进入“环境变量”界面。

• 创建用户环境变量
  
  变量名：GRAPHVIZ_INSTALL_DIR
  
  变量值：C:\Program Files (x86)\Graphviz2.34

• 创建系统环境变狼
  
  变量名：GRAPHVIZ_DOT
  
  变量值：C:\Program Files (x86)\Graphviz2.34\bin\dot.exe

• 加入系统变量 - Path
  
  双击系统变量Path, 添加路径C:\Program Files (x86)\Graphviz2.34\bin

4. 参考样例

    from keras.utils import plot_model
    from keras import models
    from keras import laters

    # Create model
    model = models.Sequential()
    model.add(layers.Dense(32, activation='relu))
    model.add(layers.Dense(1))
    model.compile(optimizer='adam', loss='mse')

    # Plot model
    plot_model(model, show_shapes=True)

注：若出现pydot can not find…

  修改pydot.py。在pydot.py中找到类Dot，修改self.prog = 'dot'为self.prog = 'dot.exe'。