keras 网络结构可视化方式

来源:互联网 发布:mindmanager注册码mac 编辑:程序博客网 时间:2024/05/17 07:07

前提:

需要安装pydot和graphviz

安装方式分别为:

pip install pydot

sudo apt-get install graphviz


安装完成后,在需要输出图像的模块中导入可视化模型函数

from keras.utils.vis_utils import plot_model


该可视化函数使用方式如下:

plot_model(model,to_file='model.png',show_shapes=True,show_layer_names=False)

原创粉丝点击