可视化深度学习网络结构图

最新推荐文章于 2024-09-12 23:40:25 发布

清风明月共天涯

最新推荐文章于 2024-09-12 23:40:25 发布

阅读量1.2k

点赞数 1

分类专栏：深度学习模型文章标签： ZF_Net 可视化模型

本文链接：https://blog.csdn.net/qq_33499064/article/details/94839110

版权

深度学习模型专栏收录该内容

2 篇文章 0 订阅

订阅专栏

欢迎大家加入京东淘宝捡垃圾群：698433653

首先，准备好你的训练模型，在这里我们以ZF_Net为例。

其次，安装可视化软件Graphviz，并设置环境变量。下载地址：https://graphviz.gitlab.io/_pages/Download/Download_windows.html

注意：下载的是它

运行安装即可。

然后，上可视化代码

from keras.utils import plot_model
from pro1.createModel import * #这里是存放的是我的训练模型
import os
os.environ["PATH"] += os.pathsep + 'E:/ruanjian/bin/'
#'E:/ruanjian/bin/' --->这里指的是可视化软件Graphviz的安装位置
norm_size = 128
depth = 3
CLASS_NUM =2
if __name__ == '__main__':
###########################################################
  # initialize the model
    print("[INFO] compiling model...")
    model = MODEL.ZF_Net(norm_size, norm_size, depth, classNum=CLASS_NUM)
    model.summary()   #这里打印模型
    plot_model(model, to_file="G:/model.png")  #绘制模型图

最后获取模型图: