利用Keras对训练好的模型结构进行可视化

一、环境的安装

在进行可视化之前需要安装一些需要用到的包
安装包请参考博客： link

二、结构可视化

编辑代码，运行。下面是对已经训练好的U-Net模型进行可视化

from keras.models import load_model
from keras.utils import plot_model

def main():
    #加载模型
    model = load_model('G:\\author_src\\unet128\\unet_src_128.h5')
    #输出模型，将结果保存到项目文件夹中
    plot_model(model, to_file='model.png', show_shapes='True')

if __name__ == '__main__':
    main()

运行结果如下图所示：

下一篇会记录度学习特征图的可视化