Layer

Filters

Ks/S1

Output Size

Encoder

Input

-

-

4 × 28 × 28

Conv1 (LeakyReLU1/Batchnorm1)

32

3 × 3/1 × 1

32 × 28 × 28

Conv2 (LeakyReLU2/Batchnorm2)

32

4 × 4/2 × 2

32 × 13 × 13

Conv3 (LeakyReLU3/Batchnorm3)

64

3 × 3/1 × 1

64 × 13 × 13

Conv4 (LeakyReLU4/Batchnorm4)

32

3 × 3/2 × 2

32 × 6 × 6

Conv5

16

3 × 3/1 × 1

16 × 6 × 6

Decoder

Input

-

-

16 × 6 × 6

ConvTrans1 (Batchnorm1/ReLU1)

32

3 × 3/1 × 1

32 × 6 × 6

ConvTrans2 (Batchnorm2/ReLU2)

64

3 × 3/2 × 2

64 × 13 × 13

ConvTrans3 (Batchnorm3/ReLU3)

32

3 × 3/1 × 1

32 × 13 × 13

ConvTrans4 (Batchnorm4/ReLU4)

32

4 × 4/2 × 2

32 × 28 × 28

ConvTrans5 (Tanh)

4

3 × 3/1 × 1

4 × 28 × 28