|
|
@@ -5,7 +5,6 @@
|
|
|
\usetikzlibrary{shapes.geometric,arrows,positioning}
|
|
|
|
|
|
\begin{document}
|
|
|
-\newcommand{\filtersize}{3}
|
|
|
|
|
|
\begin{tikzpicture}
|
|
|
\tikzstyle{conv-layer}=[draw,minimum width=3cm,minimum height=1cm]
|
|
|
@@ -22,21 +21,21 @@
|
|
|
|
|
|
% first path
|
|
|
\node (rect11) at (-2.5,-1.5) [conv-layer] {4 @ $1 \times 1 \times 256$};
|
|
|
- \node (rect12) at (-2.5,-3.0) [conv-layer] {4 @ $1 \times 1 \times 4$};
|
|
|
+ \node (rect12) at (-2.5,-3.0) [conv-layer] {4 @ $3 \times 3 \times 4$};
|
|
|
\draw[arrow] (input.south) -- (rect11.north);
|
|
|
\draw[arrow] (rect11.south) -- (rect12.north);
|
|
|
\draw[arrow] (rect12.south) -- (concatenate);
|
|
|
|
|
|
% second path
|
|
|
\node (rect21) at ( 1,-1.5) [conv-layer] {4 @ $1 \times 1 \times 256$};
|
|
|
- \node (rect22) at ( 1,-3.0) [conv-layer] {4 @ $1 \times 1 \times 4$};
|
|
|
+ \node (rect22) at ( 1,-3.0) [conv-layer] {4 @ $3 \times 3 \times 4$};
|
|
|
\draw[arrow] (input.south) -- (rect21.north);
|
|
|
\draw[arrow] (rect21.south) -- (rect22.north);
|
|
|
\draw[arrow] (rect22.south) -- (concatenate);
|
|
|
|
|
|
% last path
|
|
|
\node (rect31) at ( 5.8,-1.5) [conv-layer] {4 @ $1 \times 1 \times 256$};
|
|
|
- \node (rect32) at ( 5.8,-3.0) [conv-layer] {4 @ $1 \times 1 \times 4$};
|
|
|
+ \node (rect32) at ( 5.8,-3.0) [conv-layer] {4 @ $3 \times 3 \times 4$};
|
|
|
\draw[arrow] (input.south) -- (rect31.north);
|
|
|
\draw[arrow] (rect31.south) -- (rect32.north);
|
|
|
\draw[arrow] (rect32.south) -- (concatenate);
|
Martin Thoma
