这篇教程用图片存储和显示keras模型的结构写得很实用,希望能帮到您。 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) 1 2 3 4 就可以在指定的路径下生成模型对应的图片,下面是我resnet第一个区块所对应的图片 ———————————————— 版权声明:本文为CSDN博主「A_Z_Maps」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。 原文链接:https://blog.csdn.net/u012313895/article/details/79075750 keras 读取模型某一层的输出 深度学习(七)U-Net原理以及keras代码实现医学图像眼球血管分割 |