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

2019年4月7日 9913点热度 32人点赞 2条评论

随手画个神经网络结构!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]

Dong Wang

I will work as a PhD student of TU Graz in Austria. My research interests include Embedded/Edge AI, federated learning, computer vision, and IoT.

文章评论

  • Ryan

    你的简历吓到我了,我还在插网线

    2019年4月10日
    • nanguoyu

      @Ryan 我网页简历一年没更新了....不太正式,乱七八糟的....

      2019年4月10日
  • 此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据