【深度学习笔记】深度学习画图

PlotNeuralNet

github

Windows上使用方法

用python生成tex文件然后用TeXstudio

主要框架

import sys
sys.path.append('../')
from pycore.tikzeng import *

# defined your arch
arch = [
    to_head( '..' ),
    to_cor(),
    to_begin(),
    ###网络
    to_end()
    ]

def main():
    namefile = str(sys.argv[0]).split('.')[0]
    to_generate(arch, namefile + '.tex' )

if __name__ == '__main__':
    main()

卷积

to_Conv("conv1", 1, 1, offset="(0,0,0)", to="(0,0,0)", height=40, depth=50, width=3 )

offset

to_connection()

to_Conv("conv1", 1, 1, offset="(0,0,0)", to="(0,0,0)", height=50, depth=50, width=3 ),
to_Conv("conv2", 2, 2, offset="(6,0,0)", to="(0,0,0)", height=50, depth=50, width=3 ),
to_connection("conv1","conv2"),

to_Pool()

to_Conv("conv1", 1, 1, offset="(0,0,0)", to="(0,0,0)", height=50, depth=50, width=3 ),
to_Pool("pool1",offset="(5,0,0)",to="(conv1-east)"),
to_connection("conv1","pool1"),

to_Softmax()

to_Conv("conv1", 1, 1, offset="(0,0,0)", to="(0,0,0)", height=50, depth=50, width=3,caption