文章目录
NNSVG(个人比较常用)
http://alexlenail.me/NN-SVG/LeNet.html
这个用起来非常方便,可以画FCNN、LeNet、AlexNet格式图形,都是可以直观更改的,放个界面给大家了解一下。
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("}")
后续持续更新(2021/6/19)