神经网络结构图怎么画_怎么画高大上的神经网络结构?试试这个!
·
导读
现在这年头,做深度学习的,如果不在自己的文章里画一个神经网络结构图,都不好意思出门跟人说话。今天小编给大家介绍一个工具,让你简单又快速的搞定神经网络结构图:PlotNeuralNet。
Github 地址:
- https://github.com/HarisIqbal88/PlotNeuralNet
工具简介
PlotNeuralNet工具,具如其名,plot neural net用的,首先我们看看效果:
FCN-8
VGG-16
Holistically-Nested Edge Detection
安装使用
PlotNeuralNet的使用方法非常简单,首先将这个项目克隆下来:
git clone https://github.com/HarisIqbal88/PlotNeuralNet
然后,你可以自己写一个python脚本,想要用什么结构,就从PlotNerualNet里找对应的模块,然后,把它们拼起来就行, 好比UNet:
你只需要按部就班的堆:
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)
更多推荐
所有评论(0)