keras查看网络结构,复现别人网络结构的时候要保持尺寸的一致,用可视化的图会比较便利

      场景：在复现别人网络结构的时候,要保持网络结构的一致。每层输入的尺寸和输出的尺寸要一致。在排查复现的结构是否准确,可以通过可视化的办法。将定义好的网络结构的图可视化出来。

      keras可以很方便将网络结构可视化出来。

      

from keras.utils.visualize_util import plot
plot(model, to_file='model.png',show_shapes=True)

举个例子：可视化会对每一层的输入和输出尺寸非常直观。