【keras】用图片存储和显示keras模型的结构

原博文:

小记

最近准备在寒假结束前彻底解决掉keras的使用方法,并且在下学期结束掉tensorflow的使用,然后开始一心一意的把keras和tensorflow的源代码全部学一遍,这样才能彻掌握标准的python程序框架的写法。

keras的可视化

keras本身提供了可视化的库文件,但是由于keras的升级,已经抛弃了原来的接口和库,所以名字已经发生了改变,用网上的教程是走不通的。这里主要有三个库:

  • pydot
  • pydot_ng
  • graphviz

由于keras新版本不再使用旧的pydot接口,所以查看源代码后实际使用的是pydot_ng,原来的pydot只在异常处理的时候配合pydotplus使用。另外这里的graphviz是不能用pip安装的,安装完成后是不会识别的,一定一定要用sudo apt-get install graphviz 安装。然后在程序中使用:


from keras.utils.vis_utils import plot_model
plot_model(Mod, to_file='model1.png',show_shapes=True)

就可以在指定的路径下生成模型对应的图片,下面是我resnet第一个区块所对应的图片
MODE

猜你喜欢

转载自blog.csdn.net/u011511601/article/details/80413107