Hello, any personal favourites for drawing/visualising neural networks and transformers? With a colleague we are doing some tutorials/slides and would be very useful if there was a tool (python, latex, GUI, anything) that could help us do this, so that we can annotate on them after. Since it will be a teaching tool, clean and visually pleasing drawings would be awesome.

Would prefer a tool where we can specify the number of nodes/layers/etc and the node size and colour etc, and not something that simply draws a model from Keras/PyTorch without much adaptability.

Thanks in advance!

Comments

You must log in or register to comment.

Tober447 t1_j9oii7q wrote on February 23, 2023 at 1:24 PM

There is an old thread on reddit: https://www.reddit.com/r/MachineLearning/comments/l1z8cr/d_best_way_to_draw_neural_network_diagrams/

Personally, I like http://alexlenail.me/NN-SVG/LeNet.html

Sal-Hardin t1_j9or46r wrote on February 23, 2023 at 2:30 PM

I guess it depends on what your use-case is. How about https://github.com/HarisIqbal88/PlotNeuralNet

namuan t1_j9pagp7 wrote on February 23, 2023 at 4:40 PM

Maybe try this one . Built on top of Manim

https://github.com/helblazer811/ManimML

cumulonimbecile66 t1_j9scc8a wrote on February 24, 2023 at 5:10 AM

This is pretty cool: https://playground.tensorflow.org/

head_robotics t1_j9spe5y wrote on February 24, 2023 at 7:35 AM

This looks like it could be pretty cool:
https://github.com/lutzroeder/netron

mosquitoLad t1_j9vw4lm wrote on February 24, 2023 at 10:54 PM

Unity using the graphics encoder(C#), SDL2 + ffmpeg(C++), Ogre3D(I believe it uses C++), Blender(Python), or a mixture of the lot. You can render individual images or record a video.