画神经网络表示图

文章目录

  • NNSVG(个人比较常用)
  • DotNet

NNSVG(个人比较常用)

http://alexlenail.me/NN-SVG/LeNet.html
这个用起来非常方便,可以画FCNN、LeNet、AlexNet格式图形,都是可以直观更改的,放个界面给大家了解一下。
画神经网络表示图_第1张图片
画神经网络表示图_第2张图片
画神经网络表示图_第3张图片

DotNet

这个用起来比NNSVG稍微复制一点(需要在Linux环境中使用),需要先运行python程序,然后把输出保存为file.txt文件,然后在命令行执行dot -Tpng -O file.txt,得到图形
https://github.com/martisak/dotnets
这里我把代码复制进来了,便于大家使用。

# Inspired by
# https://tgmstat.wordpress.com/2013/06/12/draw-neural-network-diagrams-graphviz/

# UPDATE HISTORY
# April, 2018 - 2to3 - Madhavun Candadai

layers = [3, 5, 5, 5, 2]

layers_str = ["Input"] + ["Hidden"] * (len(layers) - 2) + ["Output"]
layers_col = ["none"] + ["none"] * (len(layers) - 2) + ["none"]
layers_fill = ["black"] + ["gray"] * (len(layers) - 2) + ["black"]

penwidth = 15
font = "Hilda 10"

print("digraph G {")
print("\tfontname = \"{}\"".format(font))
print("\trankdir=LR")
print("\tsplines=line")
print("\tnodesep=.08;")
print("\tranksep=1;")
print("\tedge [color=black, arrowsize=.5];")
print("\tnode [fixedsize=true,label=\"\",style=filled," + \
    "color=none,fillcolor=gray,shape=circle]\n")

# Clusters
for i in range(0, len(layers)):
    print(("\tsubgraph cluster_{} {{".format(i)))
    print(("\t\tcolor={};".format(layers_col[i])))
    print(("\t\tnode [style=filled, color=white, penwidth={},"
          "fillcolor={} shape=circle];".format(
              penwidth,
              layers_fill[i])))

    print(("\t\t"), end=' ')

    for a in range(layers[i]):
        print("l{}{} ".format(i + 1, a), end=' ')

    print(";")
    print(("\t\tlabel = {};".format(layers_str[i])))

    print("\t}\n")

# Nodes
for i in range(1, len(layers)):
    for a in range(layers[i - 1]):
        for b in range(layers[i]):
            print("\tl{}{} -> l{}{}".format(i, a, i + 1, b))

print("}")

画神经网络表示图_第4张图片


后续持续更新(2021/6/19)

你可能感兴趣的:(tensorflow,深度学习,机器学习)