杂谈 | 那些酷炫的深度学习网络图怎么画出来的?

本文我们聊聊如何才能画出炫酷高大上的神经网络图,下面是常用的几种工具。

https://zhuanlan.zhihu.com/p/60146525

https://mbd.baidu.com/newspage/data/landingshare?pageType=1&isBdboxFrom=1&context=%7B%22nid%22%3A%22news_9734687094695476916%22%2C%22sourceFrom%22%3A%22bjh%22%7D

1. NN-SVG

这个工具可以非常方便的画出各种类型的图,是下面这位小哥哥开发的,来自于麻省理工学院弗兰克尔生物工程实验室, 该实验室开发可视化和机器学习工具用于分析生物数据。

打开百度App,看更多图片


github地址:https://github.com/zfrenchee

画图工具体验地址:http://alexlenail.me/NN-SVG/

可以绘制的图包括以节点形式展示的 FCNN style,这个特别适合传统的全连接神经网络的绘制。


以平铺网络结构展示的 LeNet style,用二维的方式,适合查看每一层 featuremap 的大小和通道数目。


以三维 block 形式展现的 AlexNet style,可以更加真实地展示卷积过程中高维数据的尺度的变化,目前只支持卷积层和全连接层。


这个工具可以导出非常高清的 SVG 图,值得体验。

2. PlotNeuralNet


这个工具是萨尔大学计算机科学专业的一个学生开发的,一看就像计算机学院的嘛。

首先我们看看效果,其github链接如下,将近 4000 star:

https://github.com/HarisIqbal88/PlotNeuralNet

看看人家这个 fcn-8 的可视化图,颜值奇高。


使用的门槛相对来说就高一些了,用 LaTex 语言编辑,所以可以发挥的空间就大了,你看下面这个 softmax 层,这就是会写代码的优势了。


其中的一部分代码是这样的,会写吗。

\pic[shift={(0,0,0)}] at (0,0,0) {Box={name=crp1,caption=SoftmaxLoss: $E_\mathcal{S}$ ,%

fill={rgb:blue,1.5;red,3.5;green,3.5;white,5},opacity=0.5,height=20,width=7,depth=20}};

相似的工具还有:https://github.com/jettan/tikz_cnn

3. ConvNetDraw

ConvNetDraw 是一个使用配置命令的 CNN 神经网络画图工具,开发者是香港的一位程序员,Cédric cbovar。


采用如下的语法直接配置网络,可以简单调整 x,y,z 等 3 个维度,github 链接如下:

https://cbovar.github.io/ConvNetDraw/


使用方法如上图所示,只需输入模型结构中各层的参数配置。


挺好用的。不过它目标分辨率太低了,放大之后不清晰,达不到印刷的需求。

4. Draw_Convnet


这一个工具名叫 draw_convnet,由 Borealis 公司的员工 Gavin Weiguang Ding 提供。

简单直接,是纯用 python 代码画图的,

https://github.com/gwding/draw_convnet

看看画的图如下,核心工具是 matplotlib,图不酷炫,但是好在规规矩矩,可以严格控制,论文用挺合适的。


类似的工具还有:https://github.com/yu4u/convnet-drawer

5. Netscope

下面要说的是这个,我最常用的,caffe 的网络结构可视化工具,大名鼎鼎的 netscope,由斯坦福 AI Lab 的 Saumitro Dasgupta 开发,找不到照片就不放了,地址如下:

https://github.com/ethereon/netscope


左边放配置文件,右边出图,非常方便进行网络参数的调整和可视化。这种方式好就好在各个网络层之间的连接非常的方便。

其他

再分享一个有意思的,不是画什么正经图,但是把权重都画出来了。

http://scs.ryerson.ca/~aharley/vis/conv/





看了这么多,有人已经在偷偷笑了,上 PPT 呀,想要什么有什么,想怎么画就怎么画。



不过妹子呢?怎么不来开发一个粉色系的可视化工具呢?类似于这样的


©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • # Python 资源大全中文版 我想很多程序员应该记得 GitHub 上有一个 Awesome - XXX 系列...
    小迈克阅读 8,147评论 1 3
  • 深夜散步,深圳很美
    无法帅气阅读 1,241评论 0 0
  • 《学会写作》这本书属于可操作性强的写作指南,内容简单易读,适合不知如何上手的写作小白,可以快速摆脱“不会写”的恐惧...
    二爷酱阅读 2,713评论 0 3
  • 中里往北过三环, 步入东里休闲园。 楼群之间绿植密, 回廊敞亮广场圆。 老人场内练球技, 青年树荫恋爱谈。 几位雷...
    韩翔阅读 3,751评论 5 29
  • 决定 文/薰子 决定不回家的那一刻 开始将收拾好的行李一件件放回 折好的花裙子重新挂起来 或许本该是一场浪漫的相遇...
    會寫詩的小妮子阅读 1,001评论 0 1