NNstyle - 深度学习网络结构示意图绘制器

随手画个神经网络结构！Draw neural network architecture quickly！

一直以来，我都是用visio绘制深度学习网络结构示意图的，时常要使用visio重复操作。绘制网络机构示意图的需求很强烈，为了减轻工作量，我自己造轮子使用Pillow开发一个Python包，用来专门绘制深度学习网络结构示意图。

NNstyle是一个绘制神经网络结构的Python Package，用来快速绘制网络的结构示意图。NNstyle，支持自定义网络数量、网络颜色与透明度、图片的尺寸。NNstyle可以快速生成PNG格式的示意图片。

安装

pip install NNstyle

绘制图像

对于要绘制的每一层，我们要给出每一层网络的尺寸(height,width,depth)以及每一层的颜色(RGBA值)。

sizes = [(100, 29, 10), (100, 29, 10), (100, 5, 10), (100, 5, 10)] 
colors = [(242, 242, 242, 198), (251, 181, 128, 198), (246, 246, 246, 198), (198, 198, 198, 198)]

创建NNstyle对象

nn = NNstyle(size_layers=sizes, colors_layers=colors)
nn.makeNN()

保存图像

保存时给出保存的路径

nn.saveNN(dist="./NN.png")

示例

# import NNstyle
from NNstyle import NNstyle
sizes = [(100, 29, 10), (100, 29, 10), (100, 5, 10), (100, 5, 10)]
colors = [(242, 242, 242, 198), (251, 181, 128, 198), (246, 246, 246, 198), (198, 198, 198, 198)]
nn = NNstyle(size_layers=sizes, colors_layers=colors)
nn.makeNN()
nn.saveNN(dist="./NN.png")

这样会生成如图所示的网络结构图

下一步工作

• jpeg svg输出格式支持
• 支持网络类型输入
• 支持根据网络类型自动选择颜色
• 支持web生成，无需安装

[dm href="https://pypi.org/project/NNstyle/"]PyPI地址[/dm],[dl href="https://pypi.org/project/NNstyle/"]Pypi[/dl],[gt href="https://pypi.org/project/NNstyle/"]PyPI[/gt]