引言
- 介绍如何安装PlotNeuralNet工具，并结合chatGPT减少学习成本，快速出图。
- 将按照软件安装和软件使用，以及最终使用结果四个部分进行介绍。
正文
PlotNeuralNet
- PlotNeuralNet是一个带有python借口的latex包，能够生成任何你需要的神经网络的可视化图片。生成的图片质量很高，可以用于课程汇报，或者学术论文。
- 你可以直接使用latex编译程序，编写相关的代码，自动会生成对应的图片。我一般是使用overleaf写的latex代码，线上可以直接渲染，并下载对应的pdf文件
- overleaf链接：Overleaf
安装
- 1、在ubuntu系统中安装如下包
- Ubuntu 16.04系统
sudo apt-get install texlive-latex-extra
- Ubuntu 18.04.2输入如下指令进行安装
- 这里是需要你安装一个叫做pdflatex的软件，能够将latex的代码tex转成对应的pdf。具体安装指令如下
sudo apt-get install texlive-latex-base
sudo apt-get install texlive-fonts-recommended
sudo apt-get install texlive-fonts-extra
sudo apt-get install texlive-latex-extra
- 将latex代码文件tex后缀的，转成对应的pdf文件，具体指令见下方
pdflatex latex_source_name.tex
-
windows平台
- 下载并安装MikeTex软件，下载链接
- 下载并安装windows上运行的bash运行工具，推荐使用git的bash工具或者Cygwin
- git下载链接
- Cygwin下载链接
-
2、运行如下指令
cd pyexamples/
bash ../tikzmake.sh test_simple
-
这里可以看一下测试样例的python代码
-
完整的图片对应的latex代码
\documentclass[border=8pt, multi, tikz]{standalone}
\usepackage{import}
\subimport{../layers/}{init}
\usetikzlibrary{positioning}
\usetikzlibrary{3d} %for including external image
\def\ConvColor{rgb:yellow,5;red,2.5;white,5}
\def\ConvReluColor{rgb:yellow,5;red,5;white,5}
\def\PoolColor{rgb:red,1;black,0.3}
\def\UnpoolColor{rgb:blue,2;green,1;black,0.3}
\def\FcColor{rgb:blue,5;red,2.5;white,5}
\def\FcReluColor{rgb:blue,5;red,5;white,4}
\def\SoftmaxColor{rgb:magenta,5;black,7}
\def\SumColor{rgb:blue,5;green,15}
\newcommand{\copymidarrow}{\tikz \draw[-Stealth,line width=0.8mm,draw={rgb:blue,4;red,1;green,1;black,3}] (-0.3,0) -- ++(0.3,0);}
\begin{document}
\begin{tikzpicture}
\tikzstyle{connection}=[ultra thick,every node/.style={sloped,allow upside down},draw=\edgecolor,opacity=0.7]
\tikzstyle{copyconnection}=[ultra thick,every node/.style={sloped,allow upside down},draw={rgb:blue,4;red,1;green,1;black,3},opacity=0.7]
\pic[shift={(0,0,0)}] at (0,0,0)
{Box={
name=conv1,
caption= ,
xlabel={{64, }},
zlabel=512,
fill=\ConvColor,
height=64,
width=2,
depth=64
}
};
\pic[shift={ (0,0,0) }] at (conv1-east)
{Box={
name=pool1,
caption= ,
fill=\PoolColor,
opacity=0.5,
height=32,
width=1,
depth=32
}
};
\pic[shift={(1,0,0)}] at (pool1-east)
{Box={
name=conv2,
caption= ,
xlabel={{64, }},
zlabel=128,
fill=\ConvColor,
height=32,
width=2,
depth=32
}
};
\draw [connection] (pool1-east) -- node {\midarrow} (conv2-west);
\pic[shift={ (0,0,0) }] at (conv2-east)
{Box={
name=pool2,
caption= ,
fill=\PoolColor,
opacity=0.5,
height=28,
width=1,
depth=28
}
};
\pic[shift={(3,0,0)}] at (pool1-east)
{Box={
name=soft1,
caption=SOFT,
xlabel={{" ","dummy"}},
zlabel=10,
fill=\SoftmaxColor,
opacity=0.8,
height=3,
width=1.5,
depth=25
}
};
\draw [connection] (pool2-east) -- node {\midarrow} (soft1-west);
\pic[shift={(1.5,0,0)}] at (soft1-east)
{Ball={
name=sum1,
fill=\SumColor,
opacity=0.6,
radius=2.5,
logo=$+$
}
};
\draw [connection] (soft1-east) -- node {\midarrow} (sum1-west);
\end{tikzpicture}
\end{document}
使用
使用python进行编辑
- 具体的python编写代码
import sys
sys.path.append('../')
from pycore.tikzeng import *
# defined your arch
arch = [
to_head( '..' ),
to_cor(),
to_begin(),
to_Conv("conv1", 512, 64, offset="(0,0,0)", to="(0,0,0)", height=64, depth=64, width=2 ),
to_Pool("pool1", offset="(0,0,0)", to="(conv1-east)"),
to_Conv("conv2", 128, 64, offset="(1,0,0)", to="(pool1-east)", height=32, depth=32, width=2 ),
to_connection( "pool1", "conv2"),
to_Pool("pool2", offset="(0,0,0)", to="(conv2-east)", height=28, depth=28, width=1),
to_SoftMax("soft1", 10 ,"(3,0,0)", "(pool1-east)", caption="SOFT" ),
to_connection("pool2", "soft1"),
to_Sum("sum1", offset="(1.5,0,0)", to="(soft1-east)", radius=2.5, opacity=0.6),
to_connection("soft1", "sum1"),
to_end()
]
def main():
namefile = str(sys.argv[0]).split('.')[0]
to_generate(arch, namefile + '.tex' )
if __name__ == '__main__':
main()
- 注意切换到当前python对应的路径下方，运行如下指令
cd pyexamples/
bash ../tikzmake.sh test_simple
使用latex进行编辑
- 这里我是使用overleaf进行编辑的，也是参考了别人的模板，具体链接
- overleaf模板：链接
- 不要动layers中的包，只需要改变main.tex就可
样例
利用chatGPT
使用chatGPT生成Latex代码
- 按照网上的教程，这里需要先给他一个样例，并且这个样例和你需要画的文件具有一定的相似性，然后让chatGPT帮你生成，具体指令如下
这是生成CNN的latex代码，使用的是PlotNeuralNet的Latex包，根据这个代码，帮我写一下FCN的latex代码
- 生成效果如下
- 他生成的总归有一些问题，这个时候，你需要根据模型的实际内容进行修改。
- 我也试验了直接生成对应的模型，看了一下具体的效果
- 这里直接让他生成unet模型，它只能生成特定的模块，并不能生成完整的网络模型
- 根据具体的操作过程，在后续的过程中，会更新
利用chatGPT生成对应的python代码
- chatGPT默认的PlotNeuralNet是一个Latex包，其中并没有python的接口，所以并不能生成对应的代码。直接生成不现实。
- 这里还是尝试先给他一个代码样例，然后在提出对应的要求。效果良好。
文章来源：https://www.toymoban.com/news/detail-601149.html
- 但是，注意，PlotNeuralNet的python接口支持的组件不多，有一部分的模型画不出来，仅仅支持一下的几种
文章来源地址https://www.toymoban.com/news/detail-601149.html
总结
- 使用这个工具，虽然不能完全不需要写代码 ，但是尽可能减少了我所需要写的编码量，只需要编写局部就行了。不过还是需要能看懂，然后才能进行改错。
引用
- Creating Stunning Neural Network Visualizations with ChatGPT and PlotNeuralNet
- PlotNeuralNet
到了这里，关于论文工具——ChatGPT结合PlotNeuralNet快速出神经网络深度学习模型图的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
