tensorflow tf.keras.utils.plot_model 画深度学习神经网络拓扑图

最新推荐文章于 2021-12-03 14:00:29 发布

夏华东的博客

最新推荐文章于 2021-12-03 14:00:29 发布

阅读量2.1k

点赞数 1

文章标签： python 开发语言后端

本文链接：https://blog.csdn.net/weixin_44493841/article/details/121639152

版权

tensorflow tf.keras.utils.plot_model 画网络拓扑图

# pip install graphviz
# pip install pydot
# 下载 graphviz，并添加到环境变量。  https://graphviz.gitlab.io/download/
import tensorflow as tf
import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files/Graphviz/bin/' # graphviz的安装路径
tf.keras.utils.plot_model(
    model, # keras模型
    to_file="tensorflow 多输入 多输出 模型1.png", # 保存图片路径
    show_shapes=True, # 是否显示形状信息
    show_layer_names=True, # 是否显示图层名称
    rankdir="TB", # "TB":垂直图  "LR":水平图
    expand_nested=True, # 是否将嵌套模型展开为簇。
    dpi=96 # 图片每英寸点数。
)