117
$\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$
  • 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
80
$\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$
  • 7
    $\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
  • 1
    $\begingroup$ The tool now accommodates rectangular inputs, give it a try $\endgroup$ – David Cian Jun 20 at 14:52
47
$\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$
  • $\begingroup$ Best answer and should be voted up. $\endgroup$ – ledawg Jun 18 at 8:36
12
$\begingroup$
| improve this answer | |
$\endgroup$
  • 1
    $\begingroup$ Can VISIO draw these architectures? $\endgroup$ – Case Msee Nov 4 '16 at 6:36
2
$\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$
0
$\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$

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