DnnDraw is an editor that visualizes neural networks.
Example - tinydnn
-
Code
import dnndraw dnn = dnndraw.graph(name="tinydnn", out_format='svg') # first layer dnn.add_layer(in_layers=[], info_dict={'name': 'layer_1', 'Type': 'Conv3D', 'kernel': [1, 1, 1], 'stride': [1, 1, 1], 'padding': 'None', 'normal, relu': 'True'}) dnn.add_layer(in_layers=['layer_1'], info_dict={'name': 'layer_2', 'Type': 'Conv3D', 'kernel': [3, 3, 3], 'stride': [1, 1, 1], 'padding': 'None', 'normal, relu': 'True'}) dnn.add_layer(in_layers=['layer_1'], info_dict={'name': 'layer_3', 'Type': 'Conv3D', 'kernel': [3, 3, 3], 'stride': [1, 1, 1], 'padding': 'None', 'normal, relu': 'True'}) # end layer dnn.add_layer(in_layers=['layer_2', 'layer_3'], info_dict={'name': 'layer_4', 'Type': 'Concat'}) dnn.show() dnn.save('png')
-
Graph
Example - I3D