用 PlotNeuralNet 画神经网络的结构图

平常怎样在论文里画神经网络结构图是个头疼的问题。一直以来 PowerPoint 最受欢迎,但有没有专门的针对画神经网络结构图的工具呢?如果我要是用 Letex 怎么办,可以直接画吗?

PlotNeuralNet 就是这样一个工具,它支持 Latex,并且提供一个 Python 接口。既可以用 Latex 实现,也可以用简单的几句 Python 定义网络结构后自动生成。

图像示例

下面我们看一些例子:

FCN-8

VGG-16

Holistically-Nested Edge Detection

Python 示例

下面是怎样用 Python 定义一个简单网络结构的示例:

Latex 示例

下面是 Latex 定义 VGG-16 的代码示例:

项目 Github 地址:https://github.com/HarisIqbal88/PlotNeuralNet

PyTorch入门实战教程
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论