Keras中提供了一个神经网络可视化的函数plot_model,并可以将可视化结果保存在本地
示例效果图:
使用:
1. 安装依赖库
pip3 install pydot
#或者是pip3 install pydotplus
pip3 install graphviz
pip3 install pydot_ng
2. 调整电脑环境
将安装的graphviz的bin'文件路径加入电脑的系统环境PATH内,类似于C:/Program Files (x86)/Graphviz2.38/bin/
3. 可视化模型
from keras.utils import plot_model
plot_model(model, "model.png"); # 该函数含有其他关键字参数,可对模型可视化细节做出要求
4. 可能遇到的问题
1. 已经将依赖库安装完成,但现实找不到Graphviz包
解决:
在调用代码前手动加入以下路径:
import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'