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

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

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

相关推荐
现代野蛮人43 分钟前
【深度学习实验】—— 基于 LSTM 与 Optuna 调参的丙型肝炎预测
人工智能·深度学习·lstm
DM今天肝到几点?3 小时前
AI 安全进入「攻防同频」:Gemini 3.8 Flash Cyber 上线、Astra 触及关键级、HiddenLayer 融资 1 亿美元
网络·人工智能·深度学习·安全·语言模型·开源·知识图谱
乐迪信息3 小时前
智慧港口船舶AI算法实现在线状态监测
大数据·人工智能·深度学习·算法·计算机视觉
sel_96 小时前
深度学习损失函数详解:从 MSE、Cross Entropy 到 Dice、Focal、IoU、Contrastive Loss,一文掌握所有常见 Loss
人工智能·深度学习
147API8 小时前
蒸馏模型量化上线前,先查精度退化和算子兼容
大数据·人工智能·深度学习·机器学习·蒸馏
Omics Pro9 小时前
Arc研究所Cell|全新虚拟细胞官方基准评测框架
大数据·人工智能·深度学习·算法·机器学习
yychen_java9 小时前
从像素到汉字:基于深度学习的中文OCR系统六步全链路剖析
人工智能·深度学习·ocr
LingYi_09 小时前
建筑提取—BuildFormer
人工智能·深度学习
知识分享小能手9 小时前
深度学习学习教程,从入门到精通,概率与信息论 — 知识点详解(3)
人工智能·深度学习·学习·数据挖掘·概率论
具身AGI10 小时前
模型懂物理吗?物理AI 物理推理 的新赛点
人工智能·深度学习·计算机视觉