【深度学习】神经网络绘图可视化工具大全

一、PyTorch自带的PyTorchViz

输出的是计算图。

我的代码库中有使用示例（就是github中的使用示例）。

输出样式如下。

二、Netron

输出的是计算图。

我的代码库中有使用示例。

可以画出类似下面的图：

三、PlotNeuralNet

需要自己写 Latex code 。

可以画出类似下面的可视化图。

四、zetane

可以可视化出卷积神经网络每一层的特征图，将它们以热力图的形式展现出来。

有一个博主写的很详细：Zetane Engine ： 神经网络一键可视化_zenate engine-CSDN博客

五、TensorBoard

可以进行模型可视化。在我的博客中对土堆pytorch的总结中有讲到。

六、visualkeras

方便易用，可生成如下样式的可视化结果：

七、torchinfo

在我的代码库中，DETR就是用的这个可视化的，可生成如下类型的结果。

八、NN-SVG

是在线网站，只能画三种类型的神经网络，全连接、LeNet、AlexNet

九、Tensorspace.js

太麻烦了，不用，需要前端开发的知识。但 Tensorspace.js 的潜力很大，可生成交互式3D可视化。

可生成类似下面这样的可视化图：

十、GRAPHCORE

可画出类似下面这样的图：

大概思路是，把神经网络当成是一个图，再把这个图做图嵌入、降维或聚类，九可以得到下面这样的效果。

十一、nn_vis

和上一个类似，将神经网络的拓扑结构以3D交互的形式呈现出来

十二、PPT

万能。

十三、ProcessOn

可以绘制框图，流程图，神经网络结构图等。