CNN结构设计和可视化工具

整理盘点常用的神经网络结构可视化工具

一共盘点22个cnn可视化工具，其各有特点，我也只用过其中的两三个。

例如：

•PlotNeuralNet 常常用来做PPT里面的cnn图，看起来美观

•早期还在搞caffe的时候，就会经常使用netscope 去可视化prototxt的网络结构

•研究pytorch转onnx的时候，我就会使用 netron 去同时可视化pytorch模型和onnx模型

•在tensorflow下面，当然tesnorboard必不可少

下面的工具，大家可以各取所需

---

1、 draw_convnet^[1] 

2、 NNSVG^[2]

3、 PlotNeuralNet^[3] :

使用Latex代码来绘制神经网络结构，用在技术报告和演讲的PPT中会比较多

4、Tensorboard^[4]

大名鼎鼎，这个不用介绍了

5、 Caffe^[5]

脚本存在于 caffe/draw.py^[6]

6、 Matlab^[7]

7、Keras.js^[8]

8、net2vis^[9]

貌似对keras的api支持比较友好

9、** Netron** ^[10]

支持模型格式非常多，在做模型转换的时候非常方便

10、DotNet^[11]**

11、 Graphviz^[12] : Tutorial^[13]**

12、 Keras Visualization^[14] -

13、 Conx^[15]

这是一个python包，可以直接生成SVG, PNG, 或者 PIL 图片

14、ENNUI^[16]

可以使用交互工具来拖拽生成cnn结构

15、 NNet - R Package - Tutorial^[17]

data(infert, package="datasets")
plot(neuralnet(case~parity+induced+spontaneous, infert))

16、 GraphCore^[18]

AlexNet

ResNet50

17、 Neataptic^[19]

18、 TensorSpace^[20]:

是一个神经网络3D可视化工具，主要是tensorflow的接口

Tutorial^[21]

19、 Netscope CNN Analyzer^[22]

在线网站，主要用来可视化caffe的prototxt文件

20、Monial^[23]

Interactive Notation for Computational Graphs https://mlajtos.github.io/moniel/ 

21、 Texample^[24]

22、Quiver^[25]

大部分内容来自github，感谢ashishpatel26^[26]

References

[1] draw_convnet: https://github.com/gwding/draw_convnet
[2] NNSVG: http://alexlenail.me/NN-SVG/LeNet.html
[3] PlotNeuralNet: https://github.com/HarisIqbal88/PlotNeuralNet
[4] Tensorboard: https://www.tensorflow.org/tensorboard/graphs
[5] Caffe: https://github.com/BVLC/caffe/blob/master/python/caffe/draw.py
[6] caffe/draw.py: https://github.com/BVLC/caffe/blob/master/python/caffe/draw.py
[7] Matlab: http://www.mathworks.com/help/nnet/ref/view.html
[8] Keras.js: https://transcranial.github.io/keras-js/#/inception-v3
[9] net2vis: https://viscom.net2vis.uni-ulm.de/
[10] Netron : https://github.com/lutzroeder/Netron
[11] DotNet: https://github.com/martisak/dotnets
[12] Graphviz: http://www.graphviz.org/
[13] Tutorial: https://tgmstat.wordpress.com/2013/06/12/draw-neural-network-diagrams-graphviz/
[14] Keras Visualization: https://keras.io/visualization/
[15] Conx: https://conx.readthedocs.io/en/latest/index.html
[16] ENNUI: https://math.mit.edu/ennui/
[17] Tutorial: https://beckmw.wordpress.com/2013/03/04/visualizing-neural-networks-from-the-nnet-package/
[18] GraphCore: https://www.graphcore.ai/posts/what-does-machine-learning-look-like
[19] Neataptic: https://wagenaartje.github.io/neataptic/
[20] TensorSpace: https://tensorspace.org/
[21] Tutorial: https://www.freecodecamp.org/news/tensorspace-js-a-way-to-3d-visualize-neural-networks-in-browsers-2c0afd7648a8/
[22] Netscope CNN Analyzer: http://dgschwend.github.io/netscope/quickstart.html
[23] Monial: https://github.com/mlajtos/moniel
[24] Texample: http://www.texample.net/tikz/examples/neural-network/
[25] Quiver: https://github.com/keplr-io/quiver
[26] ashishpatel26: https://github.com/ashishpatel26/Tools-to-Design-or-Visualize-Architecture-of-Neural-Network

  ·END·

微信号：哆啦A梦爱学习

长按识别关注哦

整理不易，点个好看再走呗