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

欢迎大家加入京东淘宝捡垃圾群：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 mod