158
$\begingroup$

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

Improve this question
$\endgroup$
5
  • 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 Magalhães 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
115
$\begingroup$

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

Improve this answer
$\endgroup$
3
  • 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
  • 4
    $\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
75
$\begingroup$

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

Improve this answer
$\endgroup$
2
  • $\begingroup$ Best answer and should be voted up. $\endgroup$
    – ledawg
    Jun 18 '20 at 8:36
  • $\begingroup$ Just wanted to say that this is an amazing tool. Wish I had time to utilize this! $\endgroup$
    – rocketsfallonrocketfalls
    Jul 10 at 13:50
17
$\begingroup$
Improve this answer
$\endgroup$
1
  • 1
    $\begingroup$ Can VISIO draw these architectures? $\endgroup$
    – Case Msee
    Nov 4 '16 at 6:36
6
$\begingroup$

Netron viewer is the best tool to draw your model architecture

Image

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

Improve this answer
$\endgroup$
2
  • $\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
4
$\begingroup$

Model Architecture

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

Improve this answer
$\endgroup$
1
$\begingroup$

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

Improve this answer
$\endgroup$

Not the answer you're looking for? Browse other questions tagged or ask your own question.