I have built my model. Now I want to draw the network architecture diagram for my research paper. Example is shown below:
-
1$\begingroup$ I think graphiz, based on the dot language can, can help you. $\endgroup$ – Michael Hooreman Nov 4 '16 at 6:24
-
$\begingroup$ github.com/gwding/draw_convnet $\endgroup$ – Emre Nov 8 '16 at 3:27
-
$\begingroup$ If you are using latex, just use tikz. It does require a little time to get used to. $\endgroup$ – Ricardo Cruz Jan 25 '18 at 18:30
-
3$\begingroup$ It seems like the best approach would be to endow the network with the capability for self-reflection, and teach it how to draw its mind. Easy peasy. $\endgroup$ – Somatic Custard Jun 6 '19 at 20:33
-
$\begingroup$ For a tool & approaches overview, see: medium.com/inbrowserai/… $\endgroup$ – Piotr Migdal Jul 5 '19 at 8:51
I recently found this online tool that produces publication-ready NN-architecture schematics. It is called NN-SVG and made by Alex Lenail.
You can easily export these to use in, say, LaTeX for example.
Here are a few examples:
-
8$\begingroup$ This tool seems great -- however, I noticed that it assumes the inputs are always square. I'm in an application with a rectangular input, so this doesn't quite work. $\endgroup$ – eraoul Oct 27 '18 at 0:22
-
3$\begingroup$ The tool now accommodates rectangular inputs, give it a try $\endgroup$ – David Cian Jun 20 '20 at 14:52
-
$\begingroup$ This is great for standard CNN backbones, but in more complex architectures like encoder-decoder, attention, etc. it'd be nice to have the flexibility to re-arrange the rectangular components in any arbitrary layout $\endgroup$ – Addison Klinke Apr 1 at 19:29
I wrote some latex code to draw Deep networks for one of my reports. You can find it here: https://github.com/HarisIqbal88/PlotNeuralNet
- For automated drawing, see How do you visualize neural network architectures?, https://softwarerecs.stackexchange.com/q/28169/903 and https://softwarerecs.stackexchange.com/q/47841/903
- For manual drawing, see https://redd.it/574usi
-
1
Netron viewer is the best tool to draw your model architecture
I suppose you have a pretrained model stored in .h5 file.
-
$\begingroup$ Great tool. Thanks for sharing. This works well with MATLAB exported networks as well. $\endgroup$ – Asim Mar 9 at 7:58
-
$\begingroup$ I love Netron for visualizing and understanding my models, but I don't think it's great for explaining them to others in a paper. The diagram is far too detailed in my opinion to get a high-level overview of the architecture $\endgroup$ – Addison Klinke Apr 1 at 19:32
I drew this with draw.io, you can also choose other structures of the drawing e.g circles.
We can use Powerpoint to get the job done.
Draw the diagram (3D rectangles and perspectives come handy) -> select the interested area on the slide -> right-click -> Save as picture -> change filetype to PDF -> :)