绘制卷积神经网络

软件推荐 图表 画画 乳胶 科学计算 神经网络
2021-10-31 20:50:17

是否有任何软件用于在描述神经网络(特别是卷积网络)结构的学术论文中绘制图形?

我想要的最接近的解决方案是 TikZ LaTeX 库,它可以生成这样的图表,并使用代码描述网络(它不能处理卷积层):

tikz 神经网络 来源

描述网络结构但不以 3D 形式显示的其他软件有:

我想构建的图表遵循类似的模式,所以我很想知道是否存在比 GIMP/GraphViz/Gephi/InkScape 甚至 Powerpoint 更专业的软件来实现这一点。如果它像 TikZ 一样可编程,那就太好了。

以下是我想构建的一些图形示例(其来源如下):

在此处输入图像描述 来源

乐网5 来源

深脸 来源

3个回答

我写了一个简单的python脚本来绘制convnet,参数可调。 https://github.com/gwding/draw_convnet

draw_convnet 示例图像

如果您只需要一些简单/非花哨的插图,它可能对您有用。它复制了“基于梯度的学习应用于文档识别”中的图 2 的样式

我也为此编写了一个浏览器内工具:NN-SVG,您可以在此处找到源代码。从 FCNN 布局、LeNet 布局和 AlexNet 布局中进行选择。

在此处输入图像描述

这很棒:https ://github.com/HarisIqbal88/PlotNeuralNet 。很好地解决了我的问题,并且是用 python/tex 编写的。