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

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

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

相关推荐
机器学习之心4 分钟前
金融时间序列预测全流程框架:从SHAP特征选择到智能算法优化深度学习预测模型,核心三章实验已完成,尚未发表,期待有缘人!
人工智能·深度学习·金融
渡我白衣8 分钟前
【MySQL基础】(2):数据库基础概念
数据库·人工智能·深度学习·神经网络·mysql·机器学习·自然语言处理
【赫兹威客】浩哥13 分钟前
交通违章识别数据集与YOLO系列模型训练成果
人工智能·深度学习·机器学习
下午写HelloWorld21 分钟前
生成对抗网络GAN的简要理解
人工智能·神经网络·生成对抗网络
Yeats_Liao26 分钟前
微调决策树:何时使用Prompt Engineering,何时选择Fine-tuning?
前端·人工智能·深度学习·算法·决策树·机器学习·prompt
听麟32 分钟前
HarmonyOS 6.0+ PC端智能监控助手开发实战:摄像头联动与异常行为识别落地
人工智能·深度学习·华为·harmonyos
薛定谔的猫19829 小时前
十七、用 GPT2 中文对联模型实现经典上联自动对下联:
人工智能·深度学习·gpt2·大模型 训练 调优
陈天伟教授10 小时前
人工智能应用-机器听觉:15. 声纹识别的应用
人工智能·神经网络·机器学习·语音识别
机 _ 长11 小时前
YOLO26 改进 | 基于特征蒸馏 | 知识蒸馏 (Response & Feature-based Distillation)
python·深度学习·机器学习