前提:
需要安装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)