Keras模型可视化

通过模型可视化可以更直观的观察到搭建模型的参数和参数在不同层的传递过程。

1.安装graphviz模块,安装graphviz软件,安装pydot模块,这部分的顺序不能错,在本人的anaconda2环境中是没有自带这些模块的,但是真的需要与否还不确定。但是装上没有任何不良影响。

pip install graphviz
sudo apt-get install graphviz
pip install pydot==1.1.0 # 都说只能装1.1.0版本

2.在主函数中调用模块,首先列出几种测试不通过的版本

from keras.utils.visualize_util import plot
from keras.utils.vis_utils import plot
  • 只有下面这种是可以用的,也是官网给的(看官网API得真理)
from keras.utils import plot_model
  • 具体用法:
plot_model(temp_model, to_file='model.png',show_shapes='True')

to_file是路径,一般都保存在当前目录下
show_shapes='False'(默认),输出的是简单版的网络结构图


modelcnn.png

show_shapes='True',输出的是完全版的网络结构图

modelcnn.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容