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)