神经网络结构图怎么画_怎么画高大上的神经网络结构?试试这个!

神经网络结构图怎么画_怎么画高大上的神经网络结构?试试这个!_第1张图片

导读

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

Github 地址:

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

工具简介

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

神经网络结构图怎么画_怎么画高大上的神经网络结构?试试这个!_第2张图片

FCN-8

神经网络结构图怎么画_怎么画高大上的神经网络结构?试试这个!_第3张图片

VGG-16

神经网络结构图怎么画_怎么画高大上的神经网络结构?试试这个!_第4张图片

Holistically-Nested Edge Detection


安装使用

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

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

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

神经网络结构图怎么画_怎么画高大上的神经网络结构?试试这个!_第5张图片

你只需要按部就班的堆:

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)

你可能感兴趣的:(神经网络结构图怎么画)