How to draw Deep learning network architecture diagrams?

140

85

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

enter image description here

enter image description here

Case Msee

Posted 2016-11-03T03:10:24.893

Reputation: 1 857

1I think graphiz, based on the dot language can, can help you. – Michael Hooreman – 2016-11-04T06:24:49.487

https://github.com/gwding/draw_convnet – Emre – 2016-11-08T03:27:56.540

If you are using latex, just use tikz. It does require a little time to get used to. – Ricardo Cruz – 2018-01-25T18:30:19.127

3It 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 – 2019-06-06T20:33:35.987

For a tool & approaches overview, see: https://medium.com/inbrowserai/simple-diagrams-of-convoluted-neural-networks-39c097d2925b

– Piotr Migdal – 2019-07-05T08:51:24.070

Answers

100

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 AlexNet style


LeNet style enter image description here


and the good old Fully Connected style enter image description here

Pablo Rivas

Posted 2016-11-03T03:10:24.893

Reputation: 1 101

8This 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 – 2018-10-27T00:22:10.907

3The tool now accommodates rectangular inputs, give it a try – David Cian – 2020-06-20T14:52:43.843

62

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: enter image description here

Haris Iqbal

Posted 2016-11-03T03:10:24.893

Reputation: 629

Best answer and should be voted up. – ledawg – 2020-06-18T08:36:44.350

17

Franck Dernoncourt

Posted 2016-11-03T03:10:24.893

Reputation: 4 975

1Can VISIO draw these architectures? – Case Msee – 2016-11-04T06:36:56.940

4

Model Architecture

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

kongkip

Posted 2016-11-03T03:10:24.893

Reputation: 41

4

Netron viewer is the best tool to draw your model architecture

Image

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

oussama aatiq

Posted 2016-11-03T03:10:24.893

Reputation: 51

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

Joseph

Posted 2016-11-03T03:10:24.893

Reputation: 117