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

一、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

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

相关推荐
猎嘤一号1 小时前
使用 PyTorch 和 SwanLab 实时可视化模型训练
人工智能·pytorch·深度学习
福大大架构师每日一题2 小时前
pytorch v2.7.1 发布!全面修复关键BUG,性能与稳定性再升级,2025年深度学习利器必备!
pytorch·深度学习·bug
蹦蹦跳跳真可爱5893 小时前
Python----神经网络发(神经网络发展历程)
人工智能·python·深度学习·神经网络·计算机视觉
king of code porter4 小时前
深度学习之模型压缩三驾马车:基于ResNet18的模型剪枝实战(3)
人工智能·深度学习·剪枝
有Li4 小时前
分割任意组织:用于医学图像分割的单样本参考引导免训练自动点提示方法|文献速递-深度学习医疗AI最新文献
论文阅读·深度学习·计算机视觉
追风哥哥4 小时前
Transformer、RNN (循环神经网络) 和 CNN (卷积神经网络)的区别
rnn·深度学习·cnn·卷积神经网络·transformer
DUTBenjamin4 小时前
深度学习5——循环神经网络
人工智能·rnn·深度学习
struggle20254 小时前
tvm开源程序是适用于 CPU、GPU 和专用加速器的开放式深度学习编译器堆栈
人工智能·python·深度学习
凡人的AI工具箱4 小时前
PyTorch深度学习框架60天进阶学习计划-第57天:因果推理模型(二)- 高级算法与深度学习融合
人工智能·pytorch·深度学习·学习·mcp·a2a
硅谷秋水7 小时前
TASTE-Rob:推进面向任务的手-目标交互视频生成,实现可通用的机器人操作
人工智能·深度学习·机器学习·计算机视觉·机器人·交互