一、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
可以绘制框图,流程图,神经网络结构图等。