keras 网络结构可视化

1.安装：

conda install pygot
conda install graphviz

2.代码：（在test.py中）

导入包

from keras.utils.vis_utils import plot_model

在模型后面利用plot_model()函数，如：

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