用 PlotNeuralNet 画神经网络的结构图

PyTorch入门实战教程

平常怎样在论文里画神经网络结构图是个头疼的问题。一直以来 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入门实战教程
除特别注明外,本站所有文章均为 PyTorch 中文网原创,转载请注明出处:https://www.pytorchtutorial.com/plotneuralnet/

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部