reworked network arch diagrams

thesis/third_party/PlotNeuralNet/deepsad/arch_ef_encoder.py

@@ -25,6 +25,7 @@ arch = [
     # Input 1×32×2048  (caption carries H×W; s_filer is numeric)
     to_Conv(
         "input",
+        zlabeloffset=0.2,
         s_filer="{{2048×32}}",
         n_filer=1,
         offset="(0,0,0)",
@@ -32,7 +33,7 @@ arch = [
         height=H32,
         depth=D2048,
         width=W1,
-        caption="input",
+        caption="Input",
     ),
     # Conv1 (5x5, same): 1->8, 32×2048
     to_Conv(
@@ -48,20 +49,23 @@ arch = [
     ),
     to_Conv(
         "dwconv2",
-        s_filer="",
+        s_filer="{{2048×32}}",
+        zlabeloffset=0.15,
         n_filer=16,
         offset="(0,0,0)",
         to="(dwconv1-east)",
         height=H32,
         depth=D2048,
         width=W16,
-        caption="conv1",
+        caption="Conv1",
     ),
     # Pool1 2×2: 32×2048 -> 16×1024
     # to_connection("input", "conv1"),
     to_Pool(
         "pool1",
         offset="(0,0,0)",
+        zlabeloffset=0.3,
+        s_filer="{{512×32}}",
         to="(dwconv2-east)",
         height=H32,
         depth=D512,
@@ -82,20 +86,23 @@ arch = [
     ),
     to_Conv(
         "dwconv4",
-        s_filer="",
         n_filer=32,
+        zlabeloffset=0.3,
+        s_filer="{{512×32}}",
         offset="(0,0,0)",
         to="(dwconv3-east)",
         height=H32,
         depth=D512,
         width=W32,
-        caption="conv2",
+        caption="Conv2",
     ),
     # Pool2 2×2: 16×1024 -> 8×512
     # to_connection("pool1", "conv2"),
     to_Pool(
         "pool2",
         offset="(0,0,0)",
+        zlabeloffset=0.45,
+        s_filer="{{256×16}}",
         to="(dwconv4-east)",
         height=H16,
         depth=D256,
@@ -105,6 +112,8 @@ arch = [
     to_Pool(
         "pool3",
         offset="(0,0,0)",
+        zlabeloffset=0.45,
+        s_filer="{{128×8}}",
         to="(pool2-east)",
         height=H8,
         depth=D128,
@@ -113,19 +122,21 @@ arch = [
     ),
     to_Conv(
         "squeeze",
-        s_filer="",
         n_filer=8,
+        zlabeloffset=0.45,
+        s_filer="{{128×8}}",
         offset="(2,0,0)",
         to="(pool3-east)",
         height=H8,
         depth=D128,
         width=W8,
-        caption="squeeze",
+        caption="Squeeze",
     ),
     # FC -> rep_dim (use numeric n_filer)
     to_fc(
         "fc1",
         n_filer="{{8×128×8}}",
+        zlabeloffset=0.5,
         offset="(2,0,0)",
         to="(squeeze-east)",
         height=H1,