How to draw Deep learning network architecture diagrams?

Question

I have built my model. Now I want to draw the network architecture diagram for my research paper. Example is shown below:

I think graphiz, based on the dot language can, can help you. — Michael Hooreman, Nov 04 '16 at 06:24
If you are using latex, just use tikz. It does require a little time to get used to. — Ricardo Magalhães Cruz, Jan 25 '18 at 18:30
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. — Somatic Custard, Jun 06 '19 at 20:33
For a tool & approaches overview, see: https://medium.com/inbrowserai/simple-diagrams-of-convoluted-neural-networks-39c097d2925b — Piotr Migdal, Jul 05 '19 at 08:51

score 140 · Answer 1 · answered Aug 11 '18 at 19:05

140

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:

AlexNet style

LeNet style

and the good old Fully Connected style

answered Aug 11 '18 at 19:05

Pablo Rivas

1,501
1
9
5

9

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. – eraoul Oct 27 '18 at 00:22
5

The tool now accommodates rectangular inputs, give it a try – David Cian Jun 20 '20 at 14:52
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 – Addison Klinke Apr 01 '21 at 19:29

score 89 · Answer 2 · answered Oct 25 '18 at 17:16

89

I wrote some latex code to draw Deep networks for one of my reports. You can find it here: https://github.com/HarisIqbal88/PlotNeuralNet

With this, you can draw networks like these:

answered Oct 25 '18 at 17:16

Haris Iqbal

899
6
2

Best answer and should be voted up. – ledawg Jun 18 '20 at 08:36
Just wanted to say that this is an amazing tool. Wish I had time to utilize this! – rocketsfallonrocketfalls Jul 10 '21 at 13:50
+1. Used it for my bachelor thesis two years ago, I think it is awesome especially for slides. – Ciodar Feb 14 '23 at 21:34

Franck Dernoncourt · Answer 3 · 2017-12-29T23:52:12.190

18

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

edited Dec 29 '17 at 23:52

answered Nov 03 '16 at 03:54

Franck Dernoncourt

5,573
9
40
75

1

Can VISIO draw these architectures? – Muhammad Ali Nov 04 '16 at 06:36

score 7 · Answer 4 · edited May 12 '20 at 19:03

7

Netron viewer is the best tool to draw your model architecture

I suppose you have a pretrained model stored in .h5 file.

edited May 12 '20 at 19:03

Stephen Rauch

1,783
11
21
34

answered May 12 '20 at 17:33

oussama aatiq

89
1
1

Great tool. Thanks for sharing. This works well with MATLAB exported networks as well. – Asim Mar 09 '21 at 07:58
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 – Addison Klinke Apr 01 '21 at 19:32
the link is broken – SheppLogan May 14 '23 at 20:54

score 5 · Answer 5 · answered Jul 29 '19 at 08:28

5

I drew this with draw.io, you can also choose other structures of the drawing e.g circles.

answered Jul 29 '19 at 08:28

kongkip

51
1
2

score 1 · Answer 6 · answered Jun 06 '19 at 12:54

1

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 -> :)

answered Jun 06 '19 at 12:54

Joseph

117
3

How to draw Deep learning network architecture diagrams?

6 Answers6

Linked