I have built my model. Now I want to draw the network architecture diagram for my research paper. Example is shown below:
You can easily export these to use in, say, LaTeX for example.
Here are a few examples:
I wrote some latex code to draw Deep networks for one of my reports. You can find it here: https://github.com/HarisIqbal88/PlotNeuralNet
- For automated drawing, see How do you visualize neural network architectures?, https://softwarerecs.stackexchange.com/q/28169/903 and https://softwarerecs.stackexchange.com/q/47841/903
- For manual drawing, see https://redd.it/574usi
I drew this with draw.io, you can also choose other structures of the drawing e.g circles.
Netron viewer is the best tool to draw your model architecture
I suppose you have a pretrained model stored in .h5 file.