reworked network arch diagrams

This commit is contained in:
Jan Kowalczyk
2025-09-01 18:53:01 +02:00
parent 5ff56994c0
commit 33de01b150
17 changed files with 156 additions and 36 deletions

View File

@@ -29,6 +29,7 @@
name=input,
caption=Input,
xlabel={{1, }},
zlabeloffset=0.15,
zlabel={{2048×32}},
fill=\ConvColor,
height=26,
@@ -43,7 +44,8 @@
name=conv1,
caption=Conv1,
xlabel={{8, }},
zlabel={{1024×16}},
zlabeloffset=0.15,
zlabel={{2048×32}},
fill=\ConvColor,
height=26,
width=4,
@@ -55,6 +57,9 @@
\pic[shift={ (0,0,0) }] at (conv1-east)
{Box={
name=pool1,
xlabel={{, }},
zlabeloffset=0.3,
zlabel={{1024×16}},
caption=,
fill=\PoolColor,
opacity=0.5,
@@ -70,7 +75,8 @@
name=conv2,
caption=Conv2,
xlabel={{4, }},
zlabel={{512×8}},
zlabeloffset=0.4,
zlabel={{1024×16\hspace{2.5em}512×8}},
fill=\ConvColor,
height=18,
width=2,
@@ -82,6 +88,9 @@
\pic[shift={ (0,0,0) }] at (conv2-east)
{Box={
name=pool2,
xlabel={{, }},
zlabeloffset=0.3,
zlabel={{}},
caption=,
fill=\PoolColor,
opacity=0.5,
@@ -97,6 +106,7 @@
name=fc1,
caption=FC,
xlabel={{" ","dummy"}},
zlabeloffset=0.5,
zlabel={{4×512×8}},
fill=\FcColor,
opacity=0.8,
@@ -112,6 +122,7 @@
name=latent,
caption=Latent Space,
xlabel={{, }},
zlabeloffset=0.3,
zlabel=latent dim,
fill=\ConvColor,
height=19.200000000000003,