Drawing neural networks for publication

[juliohm]

A common task that researchers need to do is the task of drawing neural network architectures. There are various projects on the web for taking a model specification from a framework and producing a drawing like the one below:

That drawing was made with PlotNeuralNet. There is also the Netron project that supports the ONNX standard and many other model specifications:

I don't know if there is any issue open already to track this feature request, but it would be nice to take a Flux model and just export it as a LaTeX drawing made with TikZ for example.

[DrChainsaw]

You might be able to use this as a stopgap solution (sorry for self-promotion):
https://github.com/DrChainsaw/ONNXmutable.jl

I'm also genuinely interested in techniques which would allow you to do something like the above from generic julia code as the approach I used in ONNXmutable has some clear limitations.

[CarloLucibello]

An attempt to this using graphviz was done in #20 (now stale)

[davibarreira]

Any updates on this issue recently?

[DhairyaLGandhi]

There's EasyML.jl for plotting the network, but revisitng visuals with a cleaner style might be great. We can revisit the graphviz PR, a lot of it was orthogonal additions anyway, and we have more tools that we can reuse now.

[davibarreira]

I'll check EasyML.jl, thanks for the quick response! I'll also try doing some visualizations with Makie.jl :)

[DhairyaLGandhi]

Would love to see some Makie visualisations when you have them!

…

On Sat, Sep 25, 2021, 06:10 Davi Sales Barreira ***@***.***> wrote: I'll check EasyML.jl, thanks for the quick response! I'll also try doing some visualizations with Makie.jl :) — You are receiving this because you commented. Reply to this email directly, view it on GitHub <#1104 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AJOZVVPPYHWEAEBJ77NHGZTUDULBFANCNFSM4LV5ZEIA> . Triage notifications on the go with GitHub Mobile for iOS <https://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675> or Android <https://play.google.com/store/apps/details?id=com.github.android&referrer=utm_campaign%3Dnotification-email%26utm_medium%3Demail%26utm_source%3Dgithub>.