Netron viewer is the best tool to draw your model architecture

Image

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


Model Architecture

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


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


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


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


About

Geeks Mental is a community that publishes articles and tutorials about Web, Android, Data Science, new techniques and Linux security.