在学术论文中是否有任何用于绘制图形的软件来描述神经网络(特别是卷积网络)的结构?
与我想要的最接近的解决方案是TikZ LaTeX库,它可以生成像这样的图,并带有使用代码的网络(无法处理卷积层):
来源
描述网络结构但无法在3D中可视化的其他软件是:

TorchNN
TensorBoard
Mxnet

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



评论

我敢肯定,只要付出足够的努力,tikz就能生成这些图形-您可以在TeX上找到一些令人惊叹的tikz使用示例-LaTeX和texample。我可以看到输入格式可能不是您想要的格式。您能否阐明“它不能处理卷积层”的含义?

Tikz很棒,尽管我还没有找到一个例子。卷积层位只是示例中的分层正方形或矩形棱柱。

您是要以编程方式驱动图表,还是可以使用GUI界面布置图表?您还要在特定平台上使用它吗? 3D也是一个要求吗?

我需要能够可视化显示示例中的3D(无需从不同角度查看图)。 GUI很好,只要它很容易返回并删除一层或添加一层而不会花费太多时间,例如在带有A的第二个示例图中,我想删除F6和S2层,我应该可以通过删除来做到这一点,并且连接可以自行解决。

#1 楼

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



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

评论


感谢您抽出宝贵的时间编写并分享。

–詹姆斯·奥沃斯(James Owers)
16年1月1日在10:12

@gwding谢谢您,先生,这个有用的帖子。我正在尝试使用代码中的参数,但出现与索引相关的错误。您是否有更说明性的指南来更改这些参数?

–易卜拉欣·阿米尔(Ibrahim Amer)
16-9-27在0:40

@IbrahimAmer我没有任何书面说明。如果您可以将错误消息放入该仓库中的github问题中,我也许可以提供帮助

– gwding
16-9-28在14:42

@gwding使用5x5内核进行卷积时,如何获得18x18尺寸?我假设每个步骤下面的文本都指定要执行的操作以获取下一张图像,这可以通过Flatten步骤来得出结论。

– Aleksandar Jovanovic
17年8月10日在12:19



@AleksandarJovanovic的大小取决于内核大小,步幅大小和填充。我不记得确切的设置。但这可能是由于步幅大

– gwding
17年8月11日在17:31

#2 楼

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



评论


...这太棒了。感谢分享!

–詹姆斯·奥沃斯(James Owers)
18年5月10日在14:52

你在github上有源码吗?我需要分叉,谢谢:)

– Saravanabalagi Ramachandran
18年8月2日在14:53

github.com/zfrenchee/NN-SVG

– Alex Lenail
18年8月2日在19:22

#3 楼

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