Neural Network Diagram Design Tool

6 Answers 6

$\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

LeNet style enter image description here

and the good old Fully Connected style enter image description here

answered Aug 11 '18 at 19:05

Pablo RivasPablo Rivas

1,281 1 gold badge 8 silver badges 5 bronze badges

$\endgroup$

$\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$

Oct 27 '18 at 0:22
$\begingroup$ The tool now accommodates rectangular inputs, give it a try $\endgroup$

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$

Apr 1 at 19:29

$\begingroup$

answered Oct 25 '18 at 17:16

Haris IqbalHaris Iqbal

769 5 silver badges 2 bronze badges

$\endgroup$

$\begingroup$ Best answer and should be voted up. $\endgroup$

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$

Jul 10 at 13:50

$\begingroup$ Can VISIO draw these architectures? $\endgroup$

Nov 4 '16 at 6:36

$\begingroup$

Netron viewer is the best tool to draw your model architecture

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

Stephen Rauch♦

1,745 11 gold badges 17 silver badges 33 bronze badges

answered May 12 '20 at 17:33

$\endgroup$

$\begingroup$ Great tool. Thanks for sharing. This works well with MATLAB exported networks as well. $\endgroup$

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$

Apr 1 at 19:32

$\begingroup$

Model Architecture

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

answered Jul 29 '19 at 8:28

kongkipkongkip

51 1 silver badge 2 bronze badges

$\endgroup$

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

answered Jun 6 '19 at 12:54