search

HarisIqbal88 / PlotNeuralNet

Latex code for making neural networks diagrams
MIT License
21.74k stars 2.84k forks source link

Drawing the kernel dimensions in a conv layer #10

Closed mvalstar closed 5 years ago

mvalstar commented 5 years ago

Hi,

Great library, very useful. I am writing a book for Cambridge University Press and would like to use it to explain CNNs. One feature that's missing is drawing the kernel dimensions (e.g. 3x3 or 5x5 or 11x11). Do you have any suggestions how to do that?

Michel

HarisIqbal88 commented 5 years ago

Hi, Thank you for the compliment. Currently, this feature is not present in my code. However, you can use the low level tikz to do so. You might have to do something like this to draw box to represent kernel and then, label it