如何绘制深度学习网络架构图?

数据挖掘 机器学习 神经网络 深度学习 支持向量机 软件推荐
2021-10-03 19:24:28

我已经建立了我的模型。现在我想为我的研究论文绘制网络架构图。示例如下所示:

在此处输入图像描述

在此处输入图像描述

4个回答

我最近发现了这个在线工具,它可以生成可发布的 NN 架构示意图。它被称为NN-SVG,由Alex Lenail 制作

您可以轻松地导出这些以在例如LaTeX中使用。

这里有一些例子:

AlexNet 风格 AlexNet 风格


LeNet 风格 在此处输入图像描述


和旧的完全连接的风格 在此处输入图像描述

我为我的一份报告编写了一些乳胶代码来绘制深度网络。你可以在这里找到它:https ://github.com/HarisIqbal88/PlotNeuralNet

有了这个,你可以画出这样的网络: 在此处输入图像描述

Netron 查看器是绘制模型架构的最佳工具

图片

我想你有一个存储在 .h5 文件中的预训练模型。