使用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)

圖如下:

完美！！！