keras 输出网络结构_使用Keras画出网络结构图时

使用keras画神经网络结构图时容易出现：

OSError: `pydot` failed to call GraphViz.Please install GraphViz (https://www.graphviz.org/)

windows下的解决方案可以参考这篇文章：https://blog.csdn.net/sinat_40282753/article/details/85046871

在linux环境下：使用sudo apt-get install graphviz即可;

做个简单测试：

from keras.layers import Input, Convolution2D, MaxPooling2D, Flatten, Dense

from keras.models import Model

from keras.utils import plot_model

inputs = Input(shape=(229, 229, 3))

x = Convolution2D(32, 3, 3, subsample=(2, 2), border_mode='valid', dim_ordering='tf')(inputs)

x = Flatten()(x)

loss = Dense(32, activation='relu', name='loss')(x)

model = Model(input=inputs, output=loss)

model.compile(optimizer='rmsprop', loss='binary_crossentropy')

# visualize model layout with pydot_ng

plot_model(model, to_file='./model2.png', show_shapes=True)

图如下:

完美！！！