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 |