参考文章23 款神经网络的设计和可视化工具(8.12 更新) - 知乎 (zhihu.com)
这个工具是基于latex代码实现的用于绘制网络结构。效果如图
windows
首先下载并安装MikTex,下载网站知乎 - 安全中心
之后下载并安装windows的bash运行器,
安装完之后首先
cd examples/
bash ../.sh test_simple
创建新的文件夹,新建一个新的python文件,然后在这个python文件中添加代码,用于定义网络结构,主要是不同类型网络层的参数,包括输入输出数量、卷积核数量等
import sys
sys.path.append(‘../’) //这段代码修改了sys.path列表包含了python解释器自动查找所需模块的路径的字
符串,通过append方法,这里将字符串'../'添加到这个列表中,当python尝试导入一个模块时,现在也会在
当前目录的上一级目录中查找。
from pycore.tikzeng import * //这段代码的目的是从一个相对路径导入一个名为pycore的模块(或包),
并从中进一步导入tikzeng,这里pycore是一个专门用于深度学习或机器学习项目的自定义库,其中包含了各
种工具或模块,tikzeng是pycore的一个模块,用于生成TikZ代码,TikZ是一个LaTex的图形绘制包,广泛用
于生成高质量的图表和图形,在深度学习中,tikzeng用于生成神经网络架构的图形表示。
定义一个神经网络架构的列表。每一行代码都调用了一个函数来添加网络的一个组件或组件之间的关系(如连接)
arch = [
to_head('..'), //调用to_head函数来生成TikZ图的头部。
]
不想写了,还有别的活,先这样吧