105 lines
2.0 KiB
TeX
105 lines
2.0 KiB
TeX
|
|
\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);
|
|
|
|
|
|
\end{tikzpicture}
|
|
\end{document}
|
|
|