b57dcab4b3f178961548a925ce69d2bd.png

导读

现在这年头,做深度学习的,如果不在自己的文章里画一个神经网络结构图,都不好意思出门跟人说话。今天小编给大家介绍一个工具,让你简单又快速的搞定神经网络结构图:PlotNeuralNet。

Github 地址:

  • https://github.com/HarisIqbal88/PlotNeuralNet

工具简介

PlotNeuralNet工具,具如其名,plot neural net用的,首先我们看看效果:

3c72b74bbdaaf37558f2aaae8c429be2.png

FCN-8

feea5882845cc5f8592501134faa098c.png

VGG-16

c69cc6f219d4fa4d74a4b2ad980f4135.png

Holistically-Nested Edge Detection


安装使用

PlotNeuralNet的使用方法非常简单,首先将这个项目克隆下来:

git clone https://github.com/HarisIqbal88/PlotNeuralNet

然后,你可以自己写一个python脚本,想要用什么结构,就从PlotNerualNet里找对应的模块,然后,把它们拼起来就行, 好比UNet:

ff8d2a5bad1360310866d0a69084093a.png

你只需要按部就班的堆:

import syssys.path.append('../')from core.tikzeng import *from core.blocks import *arch = [  to_head('..'),  to_cor(), to_begin(),  #input to_input( '../examples/fcn8s/cats.jpg' ),  #block-001 to_ConvConvRelu( name='ccr_b1', s_filer=500, n_filer=(64,64), offset="(0,0,0)
Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐